From 580f1d439ad1e33523f263b2caaef207f7060df2 Mon Sep 17 00:00:00 2001 From: foreverpiano Date: Thu, 23 Jan 2025 13:33:04 +0000 Subject: [PATCH] upd --- fastvideo/models/hunyuan/modules/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastvideo/models/hunyuan/modules/models.py b/fastvideo/models/hunyuan/modules/models.py index 4b4c40a8c..3dcca75ca 100644 --- a/fastvideo/models/hunyuan/modules/models.py +++ b/fastvideo/models/hunyuan/modules/models.py @@ -691,7 +691,7 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin): self.x_end = img self.y_end = txt - if (EF >= block_id > SF) and counter%2 == 0 and counter < 25: + if (EF >= block_id > SF) and counter%2 == 1 and counter < 25: img = img + (self.x_end - self.x_init) else: double_block_args = [img, txt, vec, freqs_cis, text_mask] @@ -712,7 +712,7 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin): elif current_block_id == EB and counter%2 == 0: self.x_end_single = x - if (EB >= current_block_id > SB) and counter%2 == 0 and counter >= 25: + if (EB >= current_block_id > SB) and counter%2 == 1 and counter >= 25: x = x + (self.x_end_single - self.x_init_single) else: single_block_args = [x, vec, txt_seq_len, (freqs_cos, freqs_sin), text_mask]