Update gligen/ldm/models/diffusion/plms.py

gligen/ldm/models/diffusion/plms.py

@@ -205,6 +205,14 @@ class PLMSSampler(object):
             print('loss', loss,  loss2)  
             # hh = torch.autograd.backward(loss, retain_graph=True)
             grad_cond = torch.autograd.grad(loss.requires_grad_(True), [x])[0]
+            
+            del att_first
+            del att_second
+            del att_third
+            del self_first
+            del self_second
+            del self_third
+            
             # grad_cond = x.grad 
             x = x - grad_cond
             x = x.detach()