lc

2024-11-09 11:48:15 +08:00
parent 8cef82ef41
commit d39daa8254
1 changed files with 1 additions and 0 deletions
@@ -283,6 +283,7 @@ class HMVideoPipeline(StableDiffusionImg2ImgPipeline):
                pre_latents.append(pred_latent.cpu())
                control_latents.append(dicts_to_device([control_latent], device='cpu')[0])

+                progress_bar.update()
        pre_latents = torch.cat(pre_latents, dim=2)

        if post_pad_num > 0: