From e85e1ff9e6860a0a0dbb4f5a14c81567c94acc1b Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 26 Aug 2025 22:05:22 +0300
Subject: [PATCH] Update model.py

---
 wanvideo/modules/model.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index 40b25bb..88b35c9 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -1592,7 +1592,7 @@ class WanModel(torch.nn.Module):
         self.block_mask=None
 
         #S2V
-        self.zero_timestep = None
+        self.zero_timestep = self.audio_injector = None
         if cond_dim > 0:
             self.cond_encoder = nn.Conv3d(
                 cond_dim,
@@ -1619,10 +1619,9 @@ class WanModel(torch.nn.Module):
                 adain_dim=self.dim,
                 need_adain_ont=adain_mode != "attn_norm",
             )
-            self.adain_mode = adain_mode
-            self.zero_timestep = zero_timestep
-
             self.trainable_cond_mask = nn.Embedding(3, self.dim)
+        self.adain_mode = adain_mode
+        self.zero_timestep = zero_timestep
 
     @staticmethod
     def _prepare_blockwise_causal_attn_mask(