Spaces:
Sleeping
Sleeping
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
|