From 9dbd1b089b812bdb311e38b59821440738057e4a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 21 Aug 2024 12:14:39 -0700 Subject: [PATCH] fix tg ai -deprecated model --- litellm/tests/test_completion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 2ef6cf674ad..0941484d95f 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -3283,9 +3283,9 @@ def test_completion_together_ai_mixtral(): # test_completion_together_ai_mixtral() -def test_completion_together_ai_yi_chat(): +def test_completion_together_ai_llama(): litellm.set_verbose = True - model_name = "together_ai/mistralai/Mistral-7B-Instruct-v0.1" + model_name = "together_ai/meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo" try: messages = [ {"role": "user", "content": "What llm are you?"},