From 188d864fa36dbe01867ce3bf97c4bfbad36535b6 Mon Sep 17 00:00:00 2001 From: zideliu <34944964+zideliu@users.noreply.github.com> Date: Tue, 24 Oct 2023 02:38:41 +0800 Subject: [PATCH] [BUG] in transformer_temporal Fix Bugs (#5496) Fix Bugs --- src/diffusers/models/transformer_temporal.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/transformer_temporal.py b/src/diffusers/models/transformer_temporal.py index d002cb3315..d592848757 100644 --- 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() )