[BUG] in transformer_temporal Fix Bugs (#5496)

Fix Bugs

[BUG] in transformer_temporal Fix Bugs (#5496)
Fix Bugs
188d864f · zideliu · GitHub · 6e608d8a · 188d864f
Unverified Commit 188d864f authored Oct 24, 2023 by zideliu Committed by GitHub Oct 23, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/models/transformer_temporal.py src/diffusers/models/transformer_temporal.py +1 -1

No files found.
--- a/src/diffusers/models/transformer_temporal.py
+++ b/src/diffusers/models/transformer_temporal.py
@@ -171,7 +171,7 @@ class TransformerTemporalModel(ModelMixin, ConfigMixin):
        hidden_states = self.proj_out(hidden_states)
        hidden_states = (
            hidden_states[None, None, :]
-            .reshape(batch_size, height, width, channel, num_frames)
+            .reshape(batch_size, height, width, num_frames, channel)
            .permute(0, 3, 4, 1, 2)
            .contiguous()
        )