Skip to content

Commit e15afef

Browse files
committed
fixed standin modulation bug
1 parent 492fa97 commit e15afef

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

models/wan/modules/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1308,7 +1308,7 @@ def forward(
13081308
kwargs["standin_phase"] = 2
13091309
if (current_step == 0 or not standin_cache_enabled) and x_id == 0:
13101310
standin_x = self.patch_embedding(standin_ref).to(modulation_dtype).flatten(2).transpose(1, 2)
1311-
standin_e = self.time_embedding( sinusoidal_embedding_1d(self.freq_dim, torch.zeros_like(t)) )
1311+
standin_e = self.time_embedding( sinusoidal_embedding_1d(self.freq_dim, torch.zeros_like(t)).to(modulation_dtype) )
13121312
standin_e0 = self.time_projection(standin_e).unflatten(1, (6, self.dim)).to(e.dtype)
13131313
standin_e = standin_ref = None
13141314

0 commit comments

Comments
 (0)