huggingface · patrickvonplaten · Oct 11, 2023 · Oct 5, 2023 · Oct 5, 2023 · Oct 5, 2023
diff --git a/src/diffusers/models/unet_2d_blocks.py b/src/diffusers/models/unet_2d_blocks.py
@@ -466,6 +466,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.  
+
+        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 `relu`, `tanh`, and `sigmoid`.
+
+        Output:
+            A tensor with the same shape as the input tensor, but with the number of channels equal to `out_channels`.
+    """
 def get_activation(act_fn): 
 def get_activation(act_fn): 
+
     def __init__(self, in_channels: int, out_channels: int, act_fn: str):
         super().__init__()
         act_fn = get_activation(act_fn)