riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Fix type annotation for deepspeed training arg (#24988)

Browse Source
This commit is contained in:
Sylvain Gugger 2023-07-21 09:42:05 -04:00 committed by GitHub
parent 5b7ffd5492
commit a6484c89b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/training_args.py
View File

@ -1001,6 +1001,7 @@ class TrainingArguments:
)
},
)
# Do not touch this type annotation or it will stop working in CLI
fsdp_config: Optional[str] = field(
default=None,
metadata={
@ -1019,7 +1020,8 @@ class TrainingArguments:
)
},
)
deepspeed: Optional[Union[str, Dict]] = field(
# Do not touch this type annotation or it will stop working in CLI
deepspeed: Optional[str] = field(
default=None,
metadata={
"help": (