Skip to content

Commit 309f445

Browse files
committed
fix indent unet
1 parent 29d643e commit 309f445

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/diffusers/models/unet_blocks_flax.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,19 +25,19 @@ class FlaxCrossAttnDownBlock2D(nn.Module):
2525
2626
Parameters:
2727
in_channels (:obj:`int`):
28-
Input channels
28+
Input channels
2929
out_channels (:obj:`int`):
30-
Output channels
30+
Output channels
3131
dropout (:obj:`float`, *optional*, defaults to 0.0):
32-
Dropout rate
32+
Dropout rate
3333
num_layers (:obj:`int`, *optional*, defaults to 1):
34-
Number of attention blocks layers
34+
Number of attention blocks layers
3535
attn_num_head_channels (:obj:`int`, *optional*, defaults to 1):
3636
Number of attention heads of each spatial transformer block
3737
add_downsample (:obj:`bool`, *optional*, defaults to `True`):
3838
Whether to add downsampling layer before each final output
3939
dtype (:obj:`jnp.dtype`, *optional*, defaults to jnp.float32):
40-
Parameters `dtype`
40+
Parameters `dtype`
4141
"""
4242
in_channels: int
4343
out_channels: int

0 commit comments

Comments
 (0)