From e15afefb3dbf5987a1249a72c282d4e776f17a94 Mon Sep 17 00:00:00 2001 From: deepbeepmeep Date: Sat, 30 Aug 2025 04:08:22 +0200 Subject: [PATCH] fixed standin modulation bug --- models/wan/modules/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/wan/modules/model.py b/models/wan/modules/model.py index 43cf7f5..95faa4d 100644 --- a/models/wan/modules/model.py +++ b/models/wan/modules/model.py @@ -1308,7 +1308,7 @@ class WanModel(ModelMixin, ConfigMixin): kwargs["standin_phase"] = 2 if (current_step == 0 or not standin_cache_enabled) and x_id == 0: standin_x = self.patch_embedding(standin_ref).to(modulation_dtype).flatten(2).transpose(1, 2) - standin_e = self.time_embedding( sinusoidal_embedding_1d(self.freq_dim, torch.zeros_like(t)) ) + standin_e = self.time_embedding( sinusoidal_embedding_1d(self.freq_dim, torch.zeros_like(t)).to(modulation_dtype) ) standin_e0 = self.time_projection(standin_e).unflatten(1, (6, self.dim)).to(e.dtype) standin_e = standin_ref = None