Fix gradient_checkpointing_kwargs assignment in examples (#2331)

Co-authored-by: Ping <[email protected]>
huggingface · Nov 7, 2024 · ac77c09 · ac77c09
1 parent 7f2ccbe
commit ac77c09
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/scripts/dpo_online.py b/examples/scripts/dpo_online.py
@@ -65,7 +65,7 @@
 if __name__ == "__main__":
     parser = TrlParser((ScriptArguments, OnlineDPOConfig, ModelConfig))
     script_args, training_args, model_config = parser.parse_args_and_config()
-    script_args.gradient_checkpointing_kwargs = {"use_reentrant": True}
+    training_args.gradient_checkpointing_kwargs = {"use_reentrant": True}
 
     torch_dtype = (
         model_config.torch_dtype

diff --git a/examples/scripts/nash_md.py b/examples/scripts/nash_md.py
@@ -70,7 +70,7 @@
 if __name__ == "__main__":
     parser = TrlParser((ScriptArguments, NashMDConfig, ModelConfig))
     script_args, training_args, model_config = parser.parse_args_and_config()
-    script_args.gradient_checkpointing_kwargs = {"use_reentrant": True}
+    training_args.gradient_checkpointing_kwargs = {"use_reentrant": True}
 
     torch_dtype = (
         model_config.torch_dtype

diff --git a/examples/scripts/xpo.py b/examples/scripts/xpo.py
@@ -55,7 +55,7 @@
 if __name__ == "__main__":
     parser = TrlParser((ScriptArguments, XPOConfig, ModelConfig))
     script_args, training_args, model_config = parser.parse_args_and_config()
-    script_args.gradient_checkpointing_kwargs = {"use_reentrant": True}
+    training_args.gradient_checkpointing_kwargs = {"use_reentrant": True}
 
     torch_dtype = (
         model_config.torch_dtype