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

gligen/ldm/models/diffusion/plms.py

@@ -202,7 +202,7 @@ class PLMSSampler(object):
             loss2 = caculate_loss_LoCo(att_second,att_first,att_third, bboxes=bboxes,
                                 object_positions=object_positions, t = index1)*loss_scale
             loss = loss2 # + loss1
-            print('loss', loss, loss1, loss2)  
+            print('loss', loss,  loss2)  
             # hh = torch.autograd.backward(loss, retain_graph=True)
             grad_cond = torch.autograd.grad(loss.requires_grad_(True), [x])[0]
             # grad_cond = x.grad