Skip to content

Commit 12b5160

Browse files
lawrence-cjstevhliu
andcommitted
Update src/diffusers/models/transformers/sana_transformer_2d.py
Co-authored-by: Steven Liu <[email protected]>
1 parent 11f1c8d commit 12b5160

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/sana_transformer_2d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,7 @@ class SanaTransformer2DModel(ModelMixin, ConfigMixin):
303303
Parameters:
304304
num_attention_heads (int, optional, defaults to 32): The number of heads to use for multi-head attention.
305305
attention_head_dim (int, optional, defaults to 36): The number of channels in each head.
306-
in_channels (int, defaults to 4): The number of channels in the input.
306+
in_channels (int, defaults to 32): The number of channels in the input.
307307
out_channels (int, optional):
308308
The number of channels in the output. Specify this parameter if the output channel number differs from the
309309
input.

0 commit comments

Comments
 (0)