diff --git a/vlmeval/vlm/deepseek_vl.py b/vlmeval/vlm/deepseek_vl.py index edcfcae73..b420e9f5a 100644 --- a/vlmeval/vlm/deepseek_vl.py +++ b/vlmeval/vlm/deepseek_vl.py @@ -2,7 +2,7 @@ import torch from transformers import AutoModelForCausalLM import warnings -from vlmeval.smp import isimg, pip_install_robust +from vlmeval.smp import isimg class DeepSeekVL: @@ -10,8 +10,9 @@ 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)