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

Update test_hf_model.py

add_xcomposer_testcase
zhulinJulia24 2024-07-31 19:26:22 +08:00 committed by GitHub
parent a2d942b0d4
commit 128a521f6e
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

@ -23,6 +23,7 @@ class TestChat:
'model_name', 'model_name',
[ [
'internlm/internlm2_5-7b-chat', 'internlm/internlm2_5-7b-chat-1m', 'internlm/internlm2_5-7b-chat', 'internlm/internlm2_5-7b-chat-1m',
'internlm/internlm2_5-20b-chat', 'internlm/internlm2_5-1_8b-chat',
'internlm/internlm2-chat-7b', 'internlm/internlm2-chat-7b-sft', 'internlm/internlm2-chat-7b', 'internlm/internlm2-chat-7b-sft',
'internlm/internlm2-chat-20b', 'internlm/internlm2-chat-20b-sft', 'internlm/internlm2-chat-20b', 'internlm/internlm2-chat-20b-sft',
'internlm/internlm2-chat-1_8b', 'internlm/internlm2-chat-1_8b-sft' 'internlm/internlm2-chat-1_8b', 'internlm/internlm2-chat-1_8b-sft'
@ -70,7 +71,8 @@ class TestBase:
[ [
'internlm/internlm2_5-7b', 'internlm/internlm2-7b', 'internlm/internlm2_5-7b', 'internlm/internlm2-7b',
'internlm/internlm2-base-7b', 'internlm/internlm2-20b', 'internlm/internlm2-base-7b', 'internlm/internlm2-20b',
'internlm/internlm2-base-20b', 'internlm/internlm2-1_8b' 'internlm/internlm2-base-20b', 'internlm/internlm2-1_8b',
'internlm/internlm2_5-20b',
], ],
) )
@pytest.mark.parametrize( @pytest.mark.parametrize(