update args for MsDataset.load

This commit is contained in:
xingjun.wang committed 2023-12-12 13:02:54 +08:00
1 parent fe4acc66b0
commit 09533e95ed
3 files changed
+9 -1

No files matched your search

+4
View File
@@ -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