kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix a dimension bug in Transform2d (#2144)

Browse Source 
The dimension does not match when `inner_dim` is not equal to `in_channels`.
This commit is contained in:
Muyang Li
2023-02-01 04:11:45 -05:00
committed by GitHub
parent 9213d81bd0
commit 4fc7084875
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/diffusers/models/transformer_2d.py
+1 -1
View File
@@ -200,7 +200,7 @@ class Transformer2DModel(ModelMixin, ConfigMixin):
if self.is_input_continuous: if self.is_input_continuous:
# TODO: should use out_channels for continous projections # TODO: should use out_channels for continous projections
if use_linear_projection: if use_linear_projection:
self.proj_out = nn.Linear(in_channels, inner_dim) self.proj_out = nn.Linear(inner_dim, in_channels)
else: else:
self.proj_out = nn.Conv2d(inner_dim, in_channels, kernel_size=1, stride=1, padding=0) self.proj_out = nn.Conv2d(inner_dim, in_channels, kernel_size=1, stride=1, padding=0)
elif self.is_input_vectorized: elif self.is_input_vectorized: