Skip to content

Commit 64c5084

Browse files
authored
Add instruction for downloading models from openmind hub (#2577)
1 parent fd33b59 commit 64c5084

File tree

4 files changed

+13
-1
lines changed

4 files changed

+13
-1
lines changed

README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,10 @@ print(response)
199199
> By default, LMDeploy downloads model from HuggingFace. If you would like to use models from ModelScope, please install ModelScope by `pip install modelscope` and set the environment variable:
200200
>
201201
> `export LMDEPLOY_USE_MODELSCOPE=True`
202+
>
203+
> If you would like to use models from openMind Hub, please install openMind Hub by `pip install openmind_hub` and set the environment variable:
204+
>
205+
> `export LMDEPLOY_USE_OPENMIND_HUB=True`
202206
203207
For more information about inference pipeline, please refer to [here](docs/en/llm/pipeline.md).
204208

README_ja.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,10 @@ print(response)
197197
> デフォルトでは、LMDeployはHuggingFaceからモデルをダウンロードします。ModelScopeからモデルを使用する場合は、`pip install modelscope`コマンドでModelScopeをインストールし、環境変数を設定してください:
198198
>
199199
> `export LMDEPLOY_USE_MODELSCOPE=True`
200+
>
201+
> openMind Hubからモデルを使用する場合は、`pip install openmind_hub`コマンドでopenMind Hubをインストールし、環境変数を設定してください:
202+
>
203+
> `export LMDEPLOY_USE_OPENMIND_HUB=True`
200204
201205
推論パイプラインに関する詳細情報は[こちら](./docs/en/llm/pipeline.md)を参照してください。
202206

README_zh-CN.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,10 @@ print(response)
199199
> LMDeploy 默认从 HuggingFace 上面下载模型,如果要从 ModelScope 上面下载模型,请通过命令 `pip install modelscope` 安装ModelScope,并设置环境变量:
200200
>
201201
> `export LMDEPLOY_USE_MODELSCOPE=True`
202+
>
203+
> 如果要从 openMind Hub 上面下载模型,请通过命令 `pip install openmind_hub` 安装openMind Hub,并设置环境变量:
204+
>
205+
> `export LMDEPLOY_USE_OPENMIND_HUB=True`
202206
203207
关于 pipeline 的更多推理参数说明,请参考[这里](docs/zh_cn/llm/pipeline.md)
204208

lmdeploy/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def get_model(pretrained_model_name_or_path: str,
186186
download_dir: str = None,
187187
revision: str = None,
188188
token: str = None):
189-
"""Get model from huggingface or modelscope."""
189+
"""Get model from huggingface, modelscope or openmind_hub."""
190190
import os
191191
if os.getenv('LMDEPLOY_USE_MODELSCOPE', 'False').lower() == 'true':
192192
from modelscope import snapshot_download

0 commit comments

Comments
 (0)