Don't prune T2V....

2025-03-08 22:19:45 +02:00
parent 0f2c41040d
commit 15b54b92a5
1 changed files with 3 additions and 2 deletions
@@ -942,6 +942,7 @@ class HunyuanVideoPipeline(DiffusionPipeline):
                    else:
                        comfy_pbar.update(1)

-        if leapfusion_img2vid or i2v_condition_type == "latent_concat":
-            latents = latents[:, :, 1:, :, :]
+        if image_cond_latents is not None:
+            if leapfusion_img2vid or i2v_condition_type == "latent_concat":
+                latents = latents[:, :, 1:, :, :]
        return latents