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 16:21:08 +08:00 committed by GitHub
parent 9fbf7fc01c
commit dce745e5ac
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_hf_model.py
View File

@ -165,8 +165,7 @@ class TestMMModel:
'model_name',
[
'internlm/internlm-xcomposer2-7b',
'internlm/internlm-xcomposer2-7b-4bit',
'internlm/internlm-xcomposer2-4khd-7b'
'internlm/internlm-xcomposer2-7b-4bit'
],
)
def test_demo_default(self, model_name):
@ -228,7 +227,8 @@ class TestMMVlModel:
'model_name',
[
'internlm/internlm-xcomposer2-vl-7b',
'internlm/internlm-xcomposer2-vl-7b-4bit'
'internlm/internlm-xcomposer2-vl-7b-4bit',
'internlm/internlm-xcomposer2-4khd-7b'
],
)
def test_demo_default(self, model_name):