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/hparams/model_args.py
+++ b/src/llamafactory/hparams/model_args.py
@@ -267,6 +267,10 @@ class ModelArguments(QuantizationArguments, ProcessorArguments, ExportArguments,
        default=None,
        metadata={"help": "Auth token to log in with ModelScope Hub."},
    )
+    om_hub_token: Optional[str] = field(
+        default=None,
+        metadata={"help": "Auth token to log in with Modelers Hub."},
+    )
    print_param_status: bool = field(
        default=False,
        metadata={"help": "For debugging purposes, print the status of the parameters in the model."},