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

2
src/llmtuner/extras/constants.py
View File

@@ -43,7 +43,7 @@ def register_model_group(
else:
assert prefix == name.split("-")[0], "prefix should be identical."
if not os.environ.get('USE_MODELSCOPE_HUB', False):
if not int(os.environ.get('USE_MODELSCOPE_HUB', '0')):
# If path is a string, we treat it as a huggingface model-id by default.
SUPPORTED_MODELS[name] = path["hf"] if isinstance(path, dict) else path
elif isinstance(path, dict) and "ms" in path: