Skip to content

Commit cb63feb

Browse files
authored
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
1 parent 8c6b47c commit cb63feb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/models/unet_2d.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,8 @@ class UNet2DModel(ModelMixin, ConfigMixin):
4444
4545
Parameters:
4646
sample_size (`int` or `Tuple[int, int]`, *optional*, defaults to `None`):
47-
Height and width of input/output sample.
47+
Height and width of input/output sample. Dimensions must be a multiple of `2 ** (len(block_out_channels) -
48+
1)`.
4849
in_channels (`int`, *optional*, defaults to 3): Number of channels in the input image.
4950
out_channels (`int`, *optional*, defaults to 3): Number of channels in the output.
5051
center_input_sample (`bool`, *optional*, defaults to `False`): Whether to center the input sample.

0 commit comments

Comments
 (0)