From 128a521f6e7bb96e6c07d936078cbc53197bce9a Mon Sep 17 00:00:00 2001
From: zhulinJulia24 <145004780+zhulinJulia24@users.noreply.github.com>
Date: Wed, 31 Jul 2024 19:26:22 +0800
Subject: [PATCH] Update test_hf_model.py

---
 tests/test_hf_model.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py
index 513a770..126111f 100644
--- a/tests/test_hf_model.py
+++ b/tests/test_hf_model.py
@@ -23,6 +23,7 @@ class TestChat:
         'model_name',
         [
             '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-20b', 'internlm/internlm2-chat-20b-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-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(