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

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

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

@@ -200,7 +200,7 @@ class Transformer2DModel(ModelMixin, ConfigMixin):
if self.is_input_continuous:
# TODO: should use out_channels for continous projections
if use_linear_projection:
self.proj_out = nn.Linear(in_channels, inner_dim)
self.proj_out = nn.Linear(inner_dim, in_channels)
else:
self.proj_out = nn.Conv2d(inner_dim, in_channels, kernel_size=1, stride=1, padding=0)
elif self.is_input_vectorized: