From 105e4ae636afcd453ebeb57faae35f0489cb37e9 Mon Sep 17 00:00:00 2001 From: menglcai Date: Sun, 19 Jul 2026 11:04:46 +0800 Subject: [PATCH] flatten+transpose produces a non-contiguous tensor; make it contiguous before the block loop. --- src/diffusers/models/transformers/transformer_wan.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/diffusers/models/transformers/transformer_wan.py b/src/diffusers/models/transformers/transformer_wan.py index 066c9f71f3ec..6ebb5447fd16 100644 --- a/src/diffusers/models/transformers/transformer_wan.py +++ b/src/diffusers/models/transformers/transformer_wan.py @@ -670,6 +670,9 @@ def forward( hidden_states = self.patch_embedding(hidden_states) hidden_states = hidden_states.flatten(2).transpose(1, 2) + # flatten+transpose produces a non-contiguous tensor; make it contiguous before the block loop. + hidden_states = hidden_states.contiguous() + # timestep shape: batch_size, or batch_size, seq_len (wan 2.2 ti2v) if timestep.ndim == 2: ts_seq_len = timestep.shape[1]