diff --git a/litellm/tests/test_model_alias_map.py b/litellm/tests/test_model_alias_map.py index 9edd771b1..b49a9a9d8 100644 --- a/litellm/tests/test_model_alias_map.py +++ b/litellm/tests/test_model_alias_map.py @@ -14,4 +14,4 @@ litellm.set_verbose = True # Test: Check if the alias created via LiteDebugger is mapped correctly {"top_p": 0.75, "prompt": "What's the meaning of life?", "num_beams": 4, "temperature": 0.1} -print(completion("llama-7b", messages=[{"role": "user", "content": "Hey, how's it going?"}], top_p=0.1, temperature=0, num_beams=4, max_tokens=60)) \ No newline at end of file +print(completion("llama2", messages=[{"role": "user", "content": "Hey, how's it going?"}], top_p=0.1, temperature=0, num_beams=4, max_tokens=60)) \ No newline at end of file