Anonymous commited on
Commit
56cba12
1 Parent(s): 350d52b

add spaces

Browse files
lvdm/models/samplers/ddim_freetraj.py CHANGED
@@ -215,6 +215,7 @@ class DDIMSampler(object):
215
  temperature=1., noise_dropout=0., score_corrector=None, corrector_kwargs=None,
216
  unconditional_guidance_scale=1., unconditional_conditioning=None,
217
  uc_type=None, conditional_guidance_scale_temporal=None, step=0, ddim_edit=0, **kwargs):
 
218
  b, *_, device = *x.shape, x.device
219
  if x.dim() == 5:
220
  is_video = True
 
215
  temperature=1., noise_dropout=0., score_corrector=None, corrector_kwargs=None,
216
  unconditional_guidance_scale=1., unconditional_conditioning=None,
217
  uc_type=None, conditional_guidance_scale_temporal=None, step=0, ddim_edit=0, **kwargs):
218
+ print(step)
219
  b, *_, device = *x.shape, x.device
220
  if x.dim() == 5:
221
  is_video = True