Update gligen/ldm/modules/attention.py

gligen/ldm/modules/attention.py

@@ -371,10 +371,10 @@ class BasicTransformerBlock(nn.Module):
     def forward(self, x, context, objs,t):
 #        return checkpoint(self._forward, (x, context, objs), self.parameters(), self.use_checkpoint)
         # import pdb; pdb.set_trace()
-        if self.use_checkpoint and x.requires_grad:
-            return checkpoint.checkpoint(self._forward, x, context, objs,t)
-        else:
-            return self._forward(x, context, objs,t)
+        # if self.use_checkpoint and x.requires_grad:
+        #     return checkpoint.checkpoint(self._forward, x, context, objs,t)
+        # else:
+        return self._forward(x, context, objs,t)
 
     def _forward(self, x, context, objs,t): 
         # self_att_grounding = []