From 5313aa62315311b1674fd540d3abd2a7d2e37e4d Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 11 Oct 2023 11:19:35 +0000 Subject: [PATCH] make style --- src/diffusers/models/unet_2d_blocks.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/diffusers/models/unet_2d_blocks.py b/src/diffusers/models/unet_2d_blocks.py index 18b0e41af7..d57949976d 100644 --- a/src/diffusers/models/unet_2d_blocks.py +++ b/src/diffusers/models/unet_2d_blocks.py @@ -467,15 +467,18 @@ def get_up_block( class AutoencoderTinyBlock(nn.Module): """ - Tiny Autoencoder block used in [`AutoencoderTiny`]. It is a mini residual module consisting of plain conv + ReLU blocks. + Tiny Autoencoder block used in [`AutoencoderTiny`]. It is a mini residual module consisting of plain conv + ReLU + blocks. Args: in_channels (`int`): The number of input channels. out_channels (`int`): The number of output channels. - act_fn (`str`):` The activation function to use. Supported values are `"swish"`, `"mish"`, `"gelu"`, and `"relu"`. + act_fn (`str`): + ` The activation function to use. Supported values are `"swish"`, `"mish"`, `"gelu"`, and `"relu"`. Returns: - `torch.FloatTensor`: A tensor with the same shape as the input tensor, but with the number of channels equal to `out_channels`. + `torch.FloatTensor`: A tensor with the same shape as the input tensor, but with the number of channels equal to + `out_channels`. """ def __init__(self, in_channels: int, out_channels: int, act_fn: str):