diff --git a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py
index c1dfbb2..ce0bf3f 100644
--- a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py
+++ b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py
@@ -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
\ No newline at end of file