diff --git a/tests/models/automodel_test.py b/tests/models/automodel_test.py index 5888d1bd..b34955de 100644 --- a/tests/models/automodel_test.py +++ b/tests/models/automodel_test.py @@ -54,10 +54,24 @@ def _get_all_models_test_parameters(): dict(testcase_name="llama3.1-8b", model_name="llama3.1-8b"), dict(testcase_name="llama-3.1-8b", model_name="llama-3.1-8b"), dict(testcase_name="llama3.2-1b", model_name="llama3.2-1b"), - dict(testcase_name="llama3.2-1b-instruct", model_name="llama3.2-1b-instruct"), + dict( + testcase_name="llama-3.2-1b-instruct", + model_name="llama-3.2-1b-instruct", + ), + dict( + testcase_name="llama3.2-1b-instruct", + model_name="llama3.2-1b-instruct", + ), dict(testcase_name="llama-3.2-1b", model_name="llama-3.2-1b"), dict(testcase_name="llama3.2-3b", model_name="llama3.2-3b"), - dict(testcase_name="llama3.2-3b-instruct", model_name="llama3.2-3b-instruct"), + dict( + testcase_name="llama-3.2-3b-instruct", + model_name="llama-3.2-3b-instruct", + ), + dict( + testcase_name="llama3.2-3b-instruct", + model_name="llama3.2-3b-instruct", + ), dict(testcase_name="llama-3.2-3b", model_name="llama-3.2-3b"), dict(testcase_name="qwen2.5-0.5b", model_name="qwen2.5-0.5b"), dict(testcase_name="qwen2.5-1.5b", model_name="qwen2.5-1.5b"),