From 16b6fbaceccf11d521ef4b680117d2a46e927665 Mon Sep 17 00:00:00 2001 From: DeepBeepMeep Date: Wed, 26 Mar 2025 19:07:21 +0100 Subject: [PATCH] Fixe Star zero bug --- wan/image2video.py | 2 +- wan/text2video.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/wan/image2video.py b/wan/image2video.py index 67c7b07..7a9285d 100644 --- a/wan/image2video.py +++ b/wan/image2video.py @@ -409,7 +409,7 @@ class WanI2V: 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: noise_pred_uncond *= alpha noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond) diff --git a/wan/text2video.py b/wan/text2video.py index b37b383..088a9c7 100644 --- a/wan/text2video.py +++ b/wan/text2video.py @@ -306,7 +306,7 @@ class WanT2V: alpha = alpha.view(batch_size, 1, 1, 1) 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: noise_pred_uncond *= alpha noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond)