From fd20fbf4fc36dd3f5a4a38e22f16b378e5e0760e Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 23 Aug 2024 20:47:56 -0700 Subject: [PATCH] test(test_sagemaker.py): fix test --- litellm/tests/test_sagemaker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_sagemaker.py b/litellm/tests/test_sagemaker.py index 3d0d0d0cbf..3639243d3b 100644 --- a/litellm/tests/test_sagemaker.py +++ b/litellm/tests/test_sagemaker.py @@ -295,7 +295,7 @@ async def test_completion_sagemaker_prompt_template_non_stream(): ], temperature=0.2, max_tokens=80, - hf_model_name="mistralai/Mistral-7B-Instruct-v0.1", + hf_model_name="deepseek-ai/deepseek-coder-6.7b-instruct", ) # Print what was called on the mock