Skip to content

GitLab

Commit 5313aa62 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent ea8364e5
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
src/diffusers/models/unet_2d_blocks.py
View file @ 5313aa62
...@@ -467,15 +467,18 @@ def get_up_block( ...@@ -467,15 +467,18 @@ def get_up_block(
class AutoencoderTinyBlock(nn.Module): 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: Args:
in_channels (`int`): The number of input channels. in_channels (`int`): The number of input channels.
out_channels (`int`): The number of output 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: 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): def __init__(self, in_channels: int, out_channels: int, act_fn: str):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment