Skip to content

Commit 0a5a820

Browse files
committed
update
1 parent a879a22 commit 0a5a820

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -338,7 +338,7 @@ def _get_llama_prompt_embeds(
338338
image_embed_list = torch.stack(image_embed_list)
339339
image_attention_mask_list = torch.stack(image_attention_mask_list)
340340

341-
if image_embed_interleave < 6:
341+
if 0 < image_embed_interleave < 6:
342342
image_embed_list = image_embed_list[:, ::image_embed_interleave, :]
343343
image_attention_mask_list = image_attention_mask_list[:, ::image_embed_interleave]
344344

@@ -846,8 +846,9 @@ def __call__(
846846
latents = latents.to(self.vae.dtype) / self.vae.config.scaling_factor
847847
video = self.vae.decode(latents, return_dict=False)[0]
848848
video = self.video_processor.postprocess_video(video, output_type=output_type)
849+
video = video[:, :, 4:, :, :]
849850
else:
850-
video = latents
851+
video = latents[:, :, 1:, :, :]
851852

852853
# Offload all models
853854
self.maybe_free_model_hooks()

0 commit comments

Comments
 (0)