fix import error

ZX-ModelCloud · Dec 20, 2024 · 5e649d2 · 5e649d2
1 parent eca906b
commit 5e649d2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/models/test_qwen2_vl.py b/tests/models/test_qwen2_vl.py
@@ -1,4 +1,4 @@
-from qwen_vl_utils import process_vision_info
+
 
 from model_test import ModelTest
 
@@ -33,6 +33,9 @@ def test_qwen2_vl(self):
         text = processor.apply_chat_template(
             messages, tokenize=False, add_generation_prompt=True
         )
+
+        from qwen_vl_utils import process_vision_info
+
         image_inputs, video_inputs = process_vision_info(messages)
         inputs = processor(
             text=[text],