fix #2268

hiyouga · Jan 21, 2024 · e0a717a · e0a717a
1 parent 3bc301b
commit e0a717a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/llmtuner/model/patcher.py b/src/llmtuner/model/patcher.py
@@ -223,6 +223,7 @@ def _prepare_model_for_training(
             logger.warning("Current model does not support gradient checkpointing.")
         else:
             model.gradient_checkpointing_enable(gradient_checkpointing_kwargs={"use_reentrant": False})
+            model.enable_input_require_grads()
             model.config.use_cache = False  # turn off when gradient checkpointing is enabled
             logger.info("Gradient checkpointing enabled.")