nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

upd

Browse Source
This commit is contained in:
foreverpiano
2025-01-23 13:33:04 +00:00
parent f00476f51b
commit 580f1d439a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastvideo/models/hunyuan/modules/models.py
+2 -2
View File
@@ -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]