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

gligen/ldm/models/diffusion/plms.py

@@ -199,7 +199,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_att_fixed_cnt(att_second,att_first,att_third, bboxes=bboxes,
+            loss2 = caculate_loss_LoCo(att_second,att_first,att_third, bboxes=bboxes,
                                 object_positions=object_positions, t = index1)*loss_scale
             loss = loss1 + loss2
             print('loss', loss, loss1, loss2)