Update gligen/ldm/modules/diffusionmodules/openaimodel.py

gligen/ldm/modules/diffusionmodules/openaimodel.py

@@ -208,10 +208,10 @@ class ResBlock(TimestepBlock):
         # return checkpoint(
         #     self._forward, (x, emb), self.parameters(), self.use_checkpoint
         # )
-        if self.use_checkpoint and x.requires_grad:
-            return checkpoint.checkpoint(self._forward, x, emb )
-        else:
-            return self._forward(x, emb) 
+        # if self.use_checkpoint and x.requires_grad:
+        #     return checkpoint.checkpoint(self._forward, x, emb )
+        # else:
+        return self._forward(x, emb) 
 
 
     def _forward(self, x, emb):