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

gligen/ldm/models/diffusion/plms.py

@@ -197,11 +197,11 @@ class PLMSSampler(object):
             e_t,  att_first, att_second, att_third, self_first, self_second, self_third = self.model(input)
             bboxes = input['boxes_att']
             object_positions = input['object_position'] 
-            loss1 = caculate_loss_self_att(self_first, self_second, self_third, bboxes=bboxes,
-                                object_positions=object_positions, t = index1)*loss_scale 
+            # loss1 = caculate_loss_self_att(self_first, self_second, self_third, bboxes=bboxes,
+            #                     object_positions=object_positions, t = index1)*loss_scale 
             loss2 = caculate_loss_LoCo(att_second,att_first,att_third, bboxes=bboxes,
                                 object_positions=object_positions, t = index1)*loss_scale
-            loss = loss1 + loss2
+            loss = loss2 # + loss1
             print('loss', loss, loss1, loss2)  
             # hh = torch.autograd.backward(loss, retain_graph=True)
             grad_cond = torch.autograd.grad(loss.requires_grad_(True), [x])[0]