From b824888f980166a2d52c305242d4c8e44e383756 Mon Sep 17 00:00:00 2001 From: zhulinJulia24 <145004780+zhulinJulia24@users.noreply.github.com> Date: Fri, 31 May 2024 12:31:19 +0800 Subject: [PATCH] Update test_hf_model.py --- tests/test_hf_model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py index de06db3..f6e5b45 100644 --- a/tests/test_hf_model.py +++ b/tests/test_hf_model.py @@ -227,8 +227,7 @@ class TestMMVlModel: 'model_name', [ 'internlm/internlm-xcomposer2-vl-7b', - 'internlm/internlm-xcomposer2-vl-7b-4bit', - 'internlm/internlm-xcomposer2-4khd-7b' + 'internlm/internlm-xcomposer2-vl-7b-4bit' ], ) def test_demo_default(self, model_name):