update args for MsDataset.load

Former-commit-id: 09533e95ed
2026-03-08 12:46:06 +08:00 · 2023-12-12 13:02:54 +08:00
parent 9f17d36ccf
commit 879209829e
3 changed files with 9 additions and 1 deletions
--- a/src/llmtuner/hparams/model_args.py
+++ b/src/llmtuner/hparams/model_args.py
@@ -59,6 +59,10 @@ class ModelArguments:
        default=None,
        metadata={"help": "Auth token to log in with Hugging Face Hub."}
    )
+    ms_hub_token: Optional[str] = field(
+        default=None,
+        metadata={"help": "Auth token to log in with ModelScope Hub."}
+    )

    def __post_init__(self):
        self.compute_dtype = None