Add newest math model into daily testcase (InternLM#744)

llauraa23 · Jun 11, 2024 · b91949b · b91949b
1 parent 3be5894
commit b91949b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py
@@ -127,7 +127,9 @@ class TestMath:
 
     @pytest.mark.parametrize(
         'model_name',
-        ['internlm/internlm2-math-7b', 'internlm/internlm2-math-base-7b'],
+        ['internlm/internlm2-math-7b', 'internlm/internlm2-math-base-7b',
+         'internlm/internlm2-math-plus-1_8b', 'internlm/internlm2-math-plus-7b'
+        ],
     )
     @pytest.mark.parametrize(
         'usefast',