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(