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

[CI] add internlm2.5 into testcase (#751) 

Co-authored-by: zhulin1 <zhulin1@pjlab.org.cn>
internlm2-reward
zhulinJulia24 2024-07-04 19:43:21 +08:00 committed by GitHub
parent 2ebfdb900f
commit 7af5da56b9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.github/workflows/daily_tests.yaml vendored
View File

@ -7,6 +7,9 @@ env:
WORKSPACE_PREFIX: $(echo $GITHUB_WORKSPACE |cut -d '/' -f 1-4)
SLURM_PARTITION: llm_s
CONDA_BASE_ENV: internlm-model-base
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
HF_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
ACTIONS_ALLOW_USE_UNSECURE_NODE_VERSION: true
jobs:
HF_model:

14
tests/test_hf_model.py
View File

@ -21,10 +21,10 @@ class TestChat:
@pytest.mark.parametrize(
'model_name',
[
'internlm/internlm2_5-7b-chat', 'internlm/internlm2-chat-7b',
'internlm/internlm2-chat-7b-sft', 'internlm/internlm2-chat-20b',
'internlm/internlm2-chat-20b-sft', 'internlm/internlm2-chat-1_8b',
'internlm/internlm2-chat-1_8b-sft'
'internlm/internlm2_5-7b-chat', 'internlm/internlm2_5-7b-chat-1m',
'internlm/internlm2-chat-7b', 'internlm/internlm2-chat-7b-sft',
'internlm/internlm2-chat-20b', 'internlm/internlm2-chat-20b-sft',
'internlm/internlm2-chat-1_8b', 'internlm/internlm2-chat-1_8b-sft'
],
)
@pytest.mark.parametrize(
@ -84,9 +84,9 @@ class TestBase:
@pytest.mark.parametrize(
'model_name',
[
'internlm/internlm2-7b', 'internlm/internlm2-base-7b',
'internlm/internlm2-20b', 'internlm/internlm2-base-20b',
'internlm/internlm2-1_8b'
'internlm/internlm2_5-7b', 'internlm/internlm2-7b',
'internlm/internlm2-base-7b', 'internlm/internlm2-20b',
'internlm/internlm2-base-20b', 'internlm/internlm2-1_8b'
],
)
@pytest.mark.parametrize(