Update test_hf_model.py

2024-05-31 12:31:19 +08:00 · 2024-05-31 12:31:19 +08:00 · b824888f98
parent dce745e5ac
commit b824888f98
1 changed files with 1 additions and 2 deletions
--- 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):