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

gligen/ldm/models/diffusion/plms.py

@@ -209,7 +209,7 @@ class PLMSSampler(object):
             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 
-            loss2 = caculate_loss_LoCo(att_second,att_first,att_third, bboxes=bboxes,
+            loss2 = caculate_loss_LoCo_V2(att_second,att_first,att_third, bboxes=bboxes,
                                 object_positions=object_positions, t = index1)*loss_scale
             loss = loss2 # + loss1
             print('loss', loss,  loss2)