SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
1 changed files with 3 additions and 1 deletions
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( fsdp_config: Optional[str] = field(
default=None, default=None,
metadata={ 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, default=None,
metadata={ metadata={
"help": ( "help": (