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

add om_hub_token argument

Browse Source
This commit is contained in:
huniu20
2024-10-10 17:16:46 +08:00
parent 0f669f221a
commit 7b91be33c9
3 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/data/loader.py
View File

@@ -96,7 +96,7 @@ def _load_single_dataset(
data_files=data_files,
split=dataset_attr.split,
cache_dir=cache_dir,
token=model_args.ms_hub_token,
token=model_args.om_hub_token,
streaming=(data_args.streaming and (dataset_attr.load_from != "file")),
)
except ImportError: