From 355e6118d875f06fe4397a9c99edbe00ad4bd68d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 14 Jun 2025 16:46:09 -0700 Subject: [PATCH] def test_text_completion_stream(): --- tests/local_testing/test_text_completion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,