Update trainer_builder.py

axolotl-ai-cloud · Mar 16, 2024 · 34211ff · 34211ff
1 parent c7a9e76
commit 34211ff
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py
@@ -837,10 +837,6 @@ def build(self, total_num_steps):
                 training_arguments_kwargs[
                     "gradient_checkpointing_kwargs"
                 ] = self.cfg.gradient_checkpointing_kwargs
-            else:
-                training_arguments_kwargs["gradient_checkpointing_kwargs"] = {
-                    "use_reentrant": False
-                }
         if self.cfg.fsdp:
             training_arguments_kwargs["fsdp"] = self.cfg.fsdp
             if self.cfg.fsdp_config: