diff --git a/modules/sd_hijack_checkpoint.py b/modules/sd_hijack_checkpoint.py
index b2f05bbdc..2604d969f 100644
--- a/modules/sd_hijack_checkpoint.py
+++ b/modules/sd_hijack_checkpoint.py
@@ -4,19 +4,16 @@ import ldm.modules.attention
 import ldm.modules.diffusionmodules.openaimodel
 
 
-# Setting flag=False so that torch skips checking parameters.
-# parameters checking is expensive in frequent operations.
-
 def BasicTransformerBlock_forward(self, x, context=None):
-    return checkpoint(self._forward, x, context, flag=False)
+    return checkpoint(self._forward, x, context)
 
 
 def AttentionBlock_forward(self, x):
-    return checkpoint(self._forward, x, flag=False)
+    return checkpoint(self._forward, x)
 
 
 def ResBlock_forward(self, x, emb):
-    return checkpoint(self._forward, x, emb, flag=False)
+    return checkpoint(self._forward, x, emb)
 
 
 stored = []