Skip to content

Commit cddf018

Browse files
committed
fix wan recompilation issues.
1 parent 06beeca commit cddf018

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/models/transformers/transformer_wan.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -202,8 +202,8 @@ def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
202202
p_t, p_h, p_w = self.patch_size
203203
ppf, pph, ppw = num_frames // p_t, height // p_h, width // p_w
204204

205-
self.freqs = self.freqs.to(hidden_states.device)
206-
freqs = self.freqs.split_with_sizes(
205+
freqs = self.freqs.to(hidden_states.device)
206+
freqs = freqs.split_with_sizes(
207207
[
208208
self.attention_head_dim // 2 - 2 * (self.attention_head_dim // 6),
209209
self.attention_head_dim // 6,

0 commit comments

Comments
 (0)