diff --git a/tests/llm_translation/test_text_completion_unit_tests.py b/tests/llm_translation/test_text_completion_unit_tests.py index ca239ebd48..c63bf7339b 100644 --- a/tests/llm_translation/test_text_completion_unit_tests.py +++ b/tests/llm_translation/test_text_completion_unit_tests.py @@ -102,7 +102,7 @@ async def test_huggingface_text_completion_logprobs(): client = AsyncHTTPHandler() with patch.object(client, "post", return_value=return_val) as mock_post: response = await litellm.atext_completion( - model="huggingface/mistralai/Mistral-7B-v0.1", + model="huggingface/mistralai/Mistral-7B-Instruct-v0.3", prompt="good morning", client=client, ) diff --git a/tests/local_testing/test_text_completion.py b/tests/local_testing/test_text_completion.py index cf78b4b3d6..11c43de2cc 100644 --- a/tests/local_testing/test_text_completion.py +++ b/tests/local_testing/test_text_completion.py @@ -3940,7 +3940,7 @@ def test_completion_hf_prompt_array(): litellm.set_verbose = True print("\n testing hf mistral\n") response = text_completion( - model="huggingface/mistralai/Mistral-7B-v0.1", + model="huggingface/mistralai/Mistral-7B-Instruct-v0.3", prompt=token_prompt, # token prompt is a 2d list, max_tokens=0, temperature=0.0, @@ -3971,7 +3971,7 @@ def test_text_completion_stream(): try: for _ in range(2): # check if closed client used response = text_completion( - model="huggingface/mistralai/Mistral-7B-v0.1", + model="huggingface/mistralai/Mistral-7B-Instruct-v0.3", prompt="good morning", stream=True, max_tokens=10,