"tests/git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "4d1e4e24e54d00b2a1aff17410a9a86594ae8b8a"
Unverified Commit 188d864f authored by zideliu's avatar zideliu Committed by GitHub
Browse files

[BUG] in transformer_temporal Fix Bugs (#5496)

Fix Bugs
parent 6e608d8a
...@@ -171,7 +171,7 @@ class TransformerTemporalModel(ModelMixin, ConfigMixin): ...@@ -171,7 +171,7 @@ class TransformerTemporalModel(ModelMixin, ConfigMixin):
hidden_states = self.proj_out(hidden_states) hidden_states = self.proj_out(hidden_states)
hidden_states = ( hidden_states = (
hidden_states[None, None, :] 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) .permute(0, 3, 4, 1, 2)
.contiguous() .contiguous()
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment