diff --git a/lmdeploy/utils.py b/lmdeploy/utils.py index 8973ebac2..599af1590 100644 --- a/lmdeploy/utils.py +++ b/lmdeploy/utils.py @@ -190,6 +190,8 @@ def get_model(pretrained_model_name_or_path: str, import os if os.getenv('LMDEPLOY_USE_MODELSCOPE', 'False').lower() == 'true': from modelscope import snapshot_download + elif os.getenv('LMDEPLOY_USE_OPENMIND_HUB', 'False').lower() == 'true': + from openmind_hub import snapshot_download else: from huggingface_hub import snapshot_download