From 9354b3ecb3e65382e9a419d1bebcc7b5a2026222 Mon Sep 17 00:00:00 2001
From: Jixiang Hong <1751215626@qq.com>
Date: Thu, 21 Aug 2025 22:40:45 +0800
Subject: [PATCH] fix: fix the bug in
 transformers.model_utils.PretrainedModel.recompute_disable()

---
 paddlenlp/transformers/model_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddlenlp/transformers/model_utils.py b/paddlenlp/transformers/model_utils.py
index b5756896c65a..58c8514432f0 100644
--- a/paddlenlp/transformers/model_utils.py
+++ b/paddlenlp/transformers/model_utils.py
@@ -1360,8 +1360,8 @@ def recompute_disable(self):
         """
 
         def fn(layer):
-            if hasattr(layer, "enable_recompute") and (layer.enable_recompute is False or layer.enable_recompute == 0):
-                layer.enable_recompute = True
+            if hasattr(layer, "enable_recompute") and (layer.enable_recompute is True or layer.enable_recompute == 1):
+                layer.enable_recompute = False
 
         self.apply(fn)