update deepseek_vl import check

open-compass · Mar 21, 2024 · ad68b36 · ad68b36
1 parent 2c1a5ed
commit ad68b36
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/vlmeval/vlm/deepseek_vl.py b/vlmeval/vlm/deepseek_vl.py
@@ -2,16 +2,17 @@
 import torch
 from transformers import AutoModelForCausalLM
 import warnings
-from vlmeval.smp import isimg, pip_install_robust
+from vlmeval.smp import isimg
 
 
 class DeepSeekVL:
 
     INSTALL_REQ = True
 
     def check_install(self):
-        installed = pip_install_robust('deepseek_vl')
-        if not installed:
+        try:
+            import deepseek_vl
+        except ImportError:
             warnings.warn(
                 'Please first install deepseek_vl from source codes in: https://github.com/deepseek-ai/DeepSeek-VL')
             sys.exit(-1)