SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Revert "Fix typing annotations for FSDP and DeepSpeed in TrainingArguments" (#24574)

Browse Source 
Revert "Fix typing annotations for FSDP and DeepSpeed in TrainingArguments (#24549)"

This reverts commit c5e29d4381.
This commit is contained in:
Sylvain Gugger
2023-06-29 08:14:43 -04:00
committed by GitHub
parent 4f1b31c2ee
commit 2dc5e1a120
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/training_args.py
View File

@@ -976,7 +976,7 @@ class TrainingArguments:
) )
}, },
) )
fsdp_config: Optional[Union[str, Dict]] = field( fsdp_config: Optional[str] = field(
default=None, default=None,
metadata={ metadata={
"help": ( "help": (
@@ -994,11 +994,11 @@ class TrainingArguments:
) )
}, },
) )
deepspeed: Optional[Union[str, Dict]] = field( deepspeed: Optional[str] = field(
default=None, default=None,
metadata={ metadata={
"help": ( "help": (
"Enable deepspeed and pass the path to deepspeed json config file (e.g. `ds_config.json`) or an already" "Enable deepspeed and pass the path to deepspeed json config file (e.g. ds_config.json) or an already"
" loaded json file as a dict" " loaded json file as a dict"
) )
}, },