Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2026-01-12 01:03:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixe Star zero bug

Browse Source
This commit is contained in:
DeepBeepMeep 2025-03-26 19:07:21 +01:00
parent 4282a4c095
commit 16b6fbacec
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
wan/image2video.py
View File

@ -409,7 +409,7 @@ class WanI2V:
if (i <= cfg_zero_step): if (i <= cfg_zero_step):
noise_pred *= 0. # it would be faster not to compute noise_pred... noise_pred = noise_pred_text*0. # it would be faster not to compute noise_pred...
else: else:
noise_pred_uncond *= alpha noise_pred_uncond *= alpha
noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond) noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond)

2
wan/text2video.py
View File

@ -306,7 +306,7 @@ class WanT2V:
alpha = alpha.view(batch_size, 1, 1, 1) alpha = alpha.view(batch_size, 1, 1, 1)
if (i <= cfg_zero_step): if (i <= cfg_zero_step):
noise_pred *= 0. # it would be faster not to compute noise_pred... noise_pred = noise_pred_text*0. # it would be faster not to compute noise_pred...
else: else:
noise_pred_uncond *= alpha noise_pred_uncond *= alpha
noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond) noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond)