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

gligen/ldm/models/diffusion/plms.py

@@ -109,7 +109,9 @@ class PLMSSampler(object):
                     # elif loss_type=='CAR':
                     #     x = self.update_loss_only_cross( input,i, index, ts )
                     # elif loss_type=='LoCo':
-                    x = self.update_loss_LoCo( input,i, index, ts, )   
+
+                    x = self.update_loss_only_cross( input,i, index, ts )
+                    # x = self.update_loss_LoCo( input,i, index, ts, )   
                     input["x"] = x
             img, pred_x0, e_t = self.p_sample_plms(input, ts, index=index, uc=uc, guidance_scale=guidance_scale, old_eps=old_eps, t_next=ts_next)
             input["x"] = img