fix test tg - they are a very unstable provider

nidhishs · Jul 16, 2024 · 2cfafdc · 2cfafdc
1 parent 74e263b
commit 2cfafdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
@@ -2870,7 +2870,7 @@ def test_completion_together_ai_mixtral():
 
 def test_completion_together_ai_yi_chat():
  litellm.set_verbose = True
- model_name = "together_ai/zero-one-ai/Yi-34B-Chat"
+ model_name = "together_ai/mistralai/Mistral-7B-Instruct-v0.1"
  try:
  messages = [
  {"role": "user", "content": "What llm are you?"},