(chore) ci/cd testing

This commit is contained in:
ishaan-jaff 2023-11-15 18:07:01 -08:00
parent f061c775d0
commit 5990d046c4

View file

@ -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