Unverified Commit 250032e9 authored by Costa Huang's avatar Costa Huang Committed by GitHub
Browse files

Minor type annotation fix (#27276)

* Minor type annotation fix

* Trigger Build
parent a53a0c51
...@@ -1144,7 +1144,7 @@ class TrainingArguments: ...@@ -1144,7 +1144,7 @@ class TrainingArguments:
"help": "If True, use gradient checkpointing to save memory at the expense of slower backward pass." "help": "If True, use gradient checkpointing to save memory at the expense of slower backward pass."
}, },
) )
gradient_checkpointing_kwargs: dict = field( gradient_checkpointing_kwargs: Optional[dict] = field(
default=None, default=None,
metadata={ metadata={
"help": "Gradient checkpointing key word arguments such as `use_reentrant`. Will be passed to `torch.utils.checkpoint.checkpoint` through `model.gradient_checkpointing_enable`." "help": "Gradient checkpointing key word arguments such as `use_reentrant`. Will be passed to `torch.utils.checkpoint.checkpoint` through `model.gradient_checkpointing_enable`."
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment