support downloading models from openmind_hub (#2563)

InternLM · Oct 9, 2024 · a5ee8df · a5ee8df
1 parent 677207b
commit a5ee8df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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