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

gligen/ldm/models/diffusion/plms.py

@@ -152,7 +152,7 @@ class PLMSSampler(object):
             
             bboxes = input['boxes_att']
             object_positions = input['object_position'] 
-            loss2 = caculate_loss_LoCo_V2(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 = loss2
             # loss.requires_grad_(True)
@@ -165,7 +165,7 @@ class PLMSSampler(object):
             x = x - grad_cond
             x = x.detach()
             iteration += 1
-            # torch.cuda.empty_cache()
+            torch.cuda.empty_cache()
         return x
         
     def update_loss_self_cross(self, input,index1, index, ts,type_loss='self_accross' ):