Spaces:
Running
on
Zero
Running
on
Zero
Update pipeline_cogvideox_controlnet_5b_i2v_instruction2.py
Browse files
pipeline_cogvideox_controlnet_5b_i2v_instruction2.py
CHANGED
@@ -767,6 +767,8 @@ class ControlCogVideoXPipeline(DiffusionPipeline, CogVideoXLoraLoaderMixin):
|
|
767 |
|
768 |
if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
|
769 |
progress_bar.update()
|
|
|
|
|
770 |
|
771 |
if not output_type == "latent":
|
772 |
video = self.decode_latents(latents)
|
|
|
767 |
|
768 |
if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
|
769 |
progress_bar.update()
|
770 |
+
|
771 |
+
print(f"Denoising step {i}/{num_inference_steps} done.")
|
772 |
|
773 |
if not output_type == "latent":
|
774 |
video = self.decode_latents(latents)
|