From ca18bc00aaece487078d85c645d399078c93e4a6 Mon Sep 17 00:00:00 2001
From: q30056987 <985663835@qq.com>
Date: Wed, 9 Oct 2024 10:07:00 +0800
Subject: [PATCH] support downloading models from openmind_hub

---
 lmdeploy/utils.py | 2 ++
 1 file changed, 2 insertions(+)

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