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)