test(test_sagemaker.py): fix test

2024-08-23 20:47:56 -07:00 · 2024-08-23 20:47:56 -07:00 · 30ec166556
commit 30ec166556
parent f458f565af
1 changed files with 1 additions and 1 deletions
--- 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