PengWeixuanSZU commited on
Commit
78fb2f0
·
verified ·
1 Parent(s): aff8a0f

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)