diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py
index 33162f4..de06db3 100644
--- a/tests/test_hf_model.py
+++ b/tests/test_hf_model.py
@@ -165,8 +165,7 @@ class TestMMModel:
         'model_name',
         [
             'internlm/internlm-xcomposer2-7b',
-            'internlm/internlm-xcomposer2-7b-4bit',
-            'internlm/internlm-xcomposer2-4khd-7b'
+            'internlm/internlm-xcomposer2-7b-4bit'
         ],
     )
     def test_demo_default(self, model_name):
@@ -228,7 +227,8 @@ class TestMMVlModel:
         'model_name',
         [
             'internlm/internlm-xcomposer2-vl-7b',
-            'internlm/internlm-xcomposer2-vl-7b-4bit'
+            'internlm/internlm-xcomposer2-vl-7b-4bit',
+            'internlm/internlm-xcomposer2-4khd-7b'
         ],
     )
     def test_demo_default(self, model_name):