diff --git a/litellm/tests/test_model_alias_map.py b/litellm/tests/test_model_alias_map.py index 1b743b68c7f..f4647fe7cb1 100644 --- a/litellm/tests/test_model_alias_map.py +++ b/litellm/tests/test_model_alias_map.py @@ -26,7 +26,7 @@ def test_model_alias_map(): messages=[{"role": "user", "content": "Hey, how's it going?"}], top_p=0.1, temperature=0.01, - max_tokens=60, + max_tokens=10, ) print(response.model) assert "Llama-2-7b-chat-hf" in response.model