From e223fbbb9ee6931b0796243d01a38aca00d2e865 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 26 Aug 2023 10:39:50 -0700 Subject: [PATCH] update model alias map --- litellm/tests/test_model_alias_map.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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