From 7b246145689da15931626e90a637ccb240797429 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Fri, 21 Jul 2023 08:24:19 -0400 Subject: [PATCH] Fix type annotation for deepspeed training arg --- src/transformers/training_args.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py index da786317e79e6d..2efafaea77f7f0 100644 --- a/src/transformers/training_args.py +++ b/src/transformers/training_args.py @@ -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": (