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

3
tests/test_hf_model.py
View File

@ -227,8 +227,7 @@ class TestMMVlModel:
'model_name', 'model_name',
[ [
'internlm/internlm-xcomposer2-vl-7b', '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): def test_demo_default(self, model_name):