Unverified Commit 7ea6cb28 authored by Jee Jee Li's avatar Jee Jee Li Committed by GitHub
Browse files

[Misc] Improve modelscope import error (#17983)


Signed-off-by: default avatarJee Jee Li <pandaleefree@gmail.com>
parent 9fbf2bfb
...@@ -3,17 +3,21 @@ ...@@ -3,17 +3,21 @@
from vllm.envs import VLLM_USE_MODELSCOPE from vllm.envs import VLLM_USE_MODELSCOPE
if VLLM_USE_MODELSCOPE: if VLLM_USE_MODELSCOPE:
# Patch here, before each import happens try:
import modelscope # Patch here, before each import happens
from packaging import version import modelscope
from packaging import version
# patch_hub begins from modelscope>=1.18.1 # patch_hub begins from modelscope>=1.18.1
if version.parse(modelscope.__version__) <= version.parse('1.18.0'): if version.parse(modelscope.__version__) <= version.parse('1.18.0'):
raise ImportError( raise ImportError(
'Using vLLM with ModelScope needs modelscope>=1.18.1, please ' 'Using vLLM with ModelScope needs modelscope>=1.18.1, please '
'install by `pip install modelscope -U`') 'install by `pip install modelscope -U`')
from modelscope.utils.hf_util import patch_hub
from modelscope.utils.hf_util import patch_hub
# Patch hub to download models from modelscope to speed up. # Patch hub to download models from modelscope to speed up.
patch_hub() patch_hub()
except ImportError as err:
raise ImportError(
"Please install modelscope>=1.18.1 via "
"`pip install modelscope>=1.18.1` to use ModelScope.") from err
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment