From 970e9c750740c4bddb4a246f33ff84095589dc6b Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 13 Jan 2025 18:42:36 -0800 Subject: [PATCH] huggingface/mistralai/Mistral-7B-Instruct-v0.3 --- tests/llm_translation/test_text_completion_unit_tests.py | 2 +- tests/local_testing/test_text_completion.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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,