From cb63febf2ee996e2132540e119923f50780eae06 Mon Sep 17 00:00:00 2001
From: George Ogden <38294960+George-Ogden@users.noreply.github.com>
Date: Tue, 11 Apr 2023 19:02:13 +0100
Subject: [PATCH] Update documentation (#2996)

* Update documentation

Based on sampling, the width and height must be powers of 2 as the samples halve in size each time

* make style
---
 src/diffusers/models/unet_2d.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/diffusers/models/unet_2d.py b/src/diffusers/models/unet_2d.py
index d0f2a9cd8a..a83e4917c1 100644
--- a/src/diffusers/models/unet_2d.py
+++ b/src/diffusers/models/unet_2d.py
@@ -44,7 +44,8 @@ class UNet2DModel(ModelMixin, ConfigMixin):
 
     Parameters:
         sample_size (`int` or `Tuple[int, int]`, *optional*, defaults to `None`):
-            Height and width of input/output sample.
+            Height and width of input/output sample. Dimensions must be a multiple of `2 ** (len(block_out_channels) -
+            1)`.
         in_channels (`int`, *optional*, defaults to 3): Number of channels in the input image.
         out_channels (`int`, *optional*, defaults to 3): Number of channels in the output.
         center_input_sample (`bool`, *optional*, defaults to `False`): Whether to center the input sample.