Skip to content

Commit 4fc7084

Browse files
authored
Fix a dimension bug in Transform2d (huggingface#2144)
The dimension does not match when `inner_dim` is not equal to `in_channels`.
1 parent 9213d81 commit 4fc7084

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformer_2d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ def __init__(
200200
if self.is_input_continuous:
201201
# TODO: should use out_channels for continous projections
202202
if use_linear_projection:
203-
self.proj_out = nn.Linear(in_channels, inner_dim)
203+
self.proj_out = nn.Linear(inner_dim, in_channels)
204204
else:
205205
self.proj_out = nn.Conv2d(inner_dim, in_channels, kernel_size=1, stride=1, padding=0)
206206
elif self.is_input_vectorized:

0 commit comments

Comments
 (0)