We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8c6b47c commit cb63febCopy full SHA for cb63feb
src/diffusers/models/unet_2d.py
@@ -44,7 +44,8 @@ class UNet2DModel(ModelMixin, ConfigMixin):
44
45
Parameters:
46
sample_size (`int` or `Tuple[int, int]`, *optional*, defaults to `None`):
47
- 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) -
48
+ 1)`.
49
in_channels (`int`, *optional*, defaults to 3): Number of channels in the input image.
50
out_channels (`int`, *optional*, defaults to 3): Number of channels in the output.
51
center_input_sample (`bool`, *optional*, defaults to `False`): Whether to center the input sample.
0 commit comments