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

Update test_hf_model.py

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

7
tests/test_hf_model.py
View File

@ -127,7 +127,9 @@ class TestMath:
@pytest.mark.parametrize( @pytest.mark.parametrize(
'model_name', '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( @pytest.mark.parametrize(
'usefast', 'usefast',
@ -163,7 +165,8 @@ class TestMMModel:
'model_name', 'model_name',
[ [
'internlm/internlm-xcomposer2-7b', 'internlm/internlm-xcomposer2-7b',
'internlm/internlm-xcomposer2-7b-4bit' 'internlm/internlm-xcomposer2-7b-4bit',
'internlm/internlm-xcomposer2-4khd-7b'
], ],
) )
def test_demo_default(self, model_name): def test_demo_default(self, model_name):