From 4077643ce12dfdd8ea862060514d633fb9c4663c Mon Sep 17 00:00:00 2001
From: zhulin1 <zhulin1@pjlab.org.cn>
Date: Mon, 1 Jul 2024 10:56:50 +0800
Subject: [PATCH] update

---
 .github/workflows/daily_tests.yaml |  2 ++
 tests/test_hf_model.py             | 13 ++++++++-----
 2 files changed, 10 insertions(+), 5 deletions(-)

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(