diff --git a/tests/local_testing/test_text_completion.py b/tests/local_testing/test_text_completion.py index 25ec7875295..daca0b0eb76 100644 --- a/tests/local_testing/test_text_completion.py +++ b/tests/local_testing/test_text_completion.py @@ -3975,7 +3975,7 @@ def test_text_completion_stream(): try: for _ in range(2): # check if closed client used response = text_completion( - model="huggingface/mistralai/Mistral-7B-Instruct-v0.3", + model="huggingface/sarvamai/sarvam-m", prompt="good morning", stream=True, max_tokens=10,