fix test tg - they are a very unstable provider

This commit is contained in:
Ishaan Jaff 2024-07-15 21:39:28 -07:00
parent 74e263b8de
commit 2cfafdc7c3

View file

@ -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?"},