github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add newest math model into daily testcase (#744)

pull/753/head
zhulinJulia24 2024-06-11 19:29:26 +08:00 committed by GitHub
parent 3be5894976
commit b91949b918
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_hf_model.py
View File

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