File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments