AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[BUG] in transformer_temporal Fix Bugs (#5496)

Browse Source 
Fix Bugs
This commit is contained in:
zideliu
2023-10-24 02:38:41 +08:00
committed by GitHub
parent 6e608d8a35
commit 188d864fa3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/transformer_temporal.py
View File

@@ -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()
)