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-25 12:02:09 +00:00
parent 2d5001e9a0
commit b84be1662f
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastvideo/models/hunyuan/modules/models.py
+4 -3
View File
@@ -692,8 +692,9 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin):
self.x_end = img
self.y_end = txt
if (EF >= block_id > SF) and counter%2 == 0 and counter < 25:
img = img + (self.x_end - self.x_init)
if EF == block_id and counter%2 == 0 and counter < 25:
img = img + self.x_end - self.x_init
txt = txt + self.y_end - self.y_init
else:
double_block_args = [img, txt, vec, freqs_cis, text_mask]
img, txt = block(*double_block_args)
@@ -713,7 +714,7 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin):
elif current_block_id == EB and counter%2 == 1:
self.x_end_single = x
if (EB >= current_block_id > SB) and counter%2 == 0 and counter >= 25:
if EB == current_block_id and counter%2 == 0 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]