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

Fix typing annotations for FSDP and DeepSpeed in TrainingArguments (#24549)

Browse Source 
* Fix typing annotations for FSDP and DeepSpeed in TrainingArguments

* Change dict to Dict
This commit is contained in:
Max Ryabinin
2023-06-28 17:36:17 +03:00
committed by GitHub
parent daccde143d
commit c5e29d4381
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[str] = field( fsdp_config: Optional[Union[str, Dict]] = field(
default=None, default=None,
metadata={ metadata={
"help": ( "help": (
@@ -994,11 +994,11 @@ class TrainingArguments:
) )
}, },
) )
deepspeed: Optional[str] = field( deepspeed: Optional[Union[str, Dict]] = 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"
) )
}, },