Skip to content

Commit

Permalink
[BUG] in transformer_temporal Fix Bugs (huggingface#5496)
Browse files Browse the repository at this point in the history
Fix Bugs
  • Loading branch information
zideliu authored and Jimmy committed Apr 26, 2024
1 parent 97c4bd3 commit ebdd658
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/diffusers/models/transformer_temporal.py
Original file line number Diff line number Diff line change
Expand Up @@ -171,7 +171,7 @@ def forward(
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()
)
Expand Down

0 comments on commit ebdd658

Please sign in to comment.