diff --git a/.github/workflows/daily_tests.yaml b/.github/workflows/daily_tests.yaml index 488b991..b55a7ea 100644 --- a/.github/workflows/daily_tests.yaml +++ b/.github/workflows/daily_tests.yaml @@ -7,6 +7,8 @@ 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 }} jobs: HF_model: diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py index f6e5b45..dc9ed16 100644 --- a/tests/test_hf_model.py +++ b/tests/test_hf_model.py @@ -21,6 +21,7 @@ class TestChat: @pytest.mark.parametrize( '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-20b', 'internlm/internlm2-chat-20b-sft', 'internlm/internlm2-chat-1_8b', 'internlm/internlm2-chat-1_8b-sft' @@ -83,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( @@ -127,8 +128,10 @@ class TestMath: @pytest.mark.parametrize( '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(