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)
|