423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 03:10:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

add readme

Browse Source
This commit is contained in:
yuze.zyz
2023-12-01 16:11:30 +08:00
parent 8ce4d11e38
commit 5aa6751e52
4 changed files with 61 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/model/loader.py
View File

@@ -235,7 +235,7 @@ def load_model_and_tokenizer(
def try_download_model_from_ms(model_args):
if os.environ.get('USE_MODELSCOPE_HUB', False) and not os.path.exists(model_args.model_name_or_path):
if int(os.environ.get('USE_MODELSCOPE_HUB', '0')) and not os.path.exists(model_args.model_name_or_path):
try:
from modelscope import snapshot_download
revision = model_args.model_revision
@@ -243,5 +243,5 @@ def try_download_model_from_ms(model_args):
revision = 'master'
model_args.model_name_or_path = snapshot_download(model_args.model_name_or_path, revision)
except ImportError as e:
raise ImportError(f'You are using `USE_MODELSCOPE_HUB=True` but you have no modelscope sdk installed. '
raise ImportError(f'You are using `USE_MODELSCOPE_HUB=1` but you have no modelscope sdk installed. '
f'Please install it by `pip install modelscope -U`') from e