AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Update documentation (#2996)

Browse Source 
* Update documentation

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

* make style
This commit is contained in:
George Ogden
2023-04-11 19:02:13 +01:00
committed by GitHub
parent 8c6b47cfde
commit cb63febf2e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/diffusers/models/unet_2d.py
View File

@@ -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.