add om_hub_token argument

Former-commit-id: 7b91be33c9
2025-12-17 20:30:36 +08:00 · 2024-10-10 17:16:46 +08:00
parent 132c1f1b0f
commit e8e98bb125
3 changed files with 6 additions and 2 deletions
--- a/src/llamafactory/data/loader.py
+++ b/src/llamafactory/data/loader.py
@@ -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: