Pusheen commited on
Commit
f4f89f9
·
verified ·
1 Parent(s): d053d42

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

Browse files
gligen/ldm/models/diffusion/plms.py CHANGED
@@ -108,7 +108,7 @@ class PLMSSampler(object):
108
  # x = self.update_only_self( input,i, index, ts )
109
  # elif loss_type=='CAR':
110
  # x = self.update_loss_only_cross( input,i, index, ts )
111
- x = self.update_loss_SAR_CAR( input,i, index, ts )
112
  input["x"] = x
113
  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)
114
  input["x"] = img
 
108
  # x = self.update_only_self( input,i, index, ts )
109
  # elif loss_type=='CAR':
110
  # x = self.update_loss_only_cross( input,i, index, ts )
111
+ x = self.update_loss_self_cross( input,i, index, ts )
112
  input["x"] = x
113
  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)
114
  input["x"] = img