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

update

pull/751/head
zhulin1 2024-07-01 10:56:50 +08:00
parent 9943444614
commit 4077643ce1
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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

13
tests/test_hf_model.py
View File

@ -21,6 +21,7 @@ class TestChat:
@pytest.mark.parametrize( @pytest.mark.parametrize(
'model_name', 'model_name',
[ [
'internlm/internlm2_5-7b-chat', 'internlm/internlm2_5-7b-chat-1m',
'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'
@ -83,9 +84,9 @@ class TestBase:
@pytest.mark.parametrize( @pytest.mark.parametrize(
'model_name', 'model_name',
[ [
'internlm/internlm2-7b', 'internlm/internlm2-base-7b', 'internlm/internlm2_5-7b', 'internlm/internlm2-7b',
'internlm/internlm2-20b', 'internlm/internlm2-base-20b', 'internlm/internlm2-base-7b', 'internlm/internlm2-20b',
'internlm/internlm2-1_8b' 'internlm/internlm2-base-20b', 'internlm/internlm2-1_8b'
], ],
) )
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -127,8 +128,10 @@ class TestMath:
@pytest.mark.parametrize( @pytest.mark.parametrize(
'model_name', 'model_name',
['internlm/internlm2-math-7b', 'internlm/internlm2-math-base-7b', [
'internlm/internlm2-math-plus-1_8b', 'internlm/internlm2-math-plus-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(