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

bugs fixed

Browse Source 
Former-commit-id: 843b5d85e9
This commit is contained in:
huniu20
2024-10-11 19:56:13 +08:00
parent e8e98bb125
commit 5a3280ebee
2 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/extras/misc.py
View File

@@ -237,11 +237,9 @@ def try_download_model_from_other_hub(model_args: "ModelArguments") -> str:
if use_openmind():
try:
import openmind
from openmind.utils.hub import snapshot_download
revision = "main" if model_args.model_revision == "main" else model_args.model_revision
return snapshot_download(model_args.model_name_or_path, revision=revision, cache_dir=model_args.cache_dir)
return snapshot_download(model_args.model_name_or_path, revision=model_args.model_revision, cache_dir=model_args.cache_dir)
except ImportError:
raise ImportError("Please install openmind and openmind_hub via `pip install openmind -U`")