Skip to content

Commit 57525bb

Browse files
authored
Fix documentation typo for UNet2DModel and UNet2DConditionModel (huggingface#1275)
* Fix documentation typo * Fix other typo
1 parent 7c5fef8 commit 57525bb

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/diffusers/models/unet_2d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ class UNet2DModel(ModelMixin, ConfigMixin):
5151
time_embedding_type (`str`, *optional*, defaults to `"positional"`): Type of time embedding to use.
5252
freq_shift (`int`, *optional*, defaults to 0): Frequency shift for fourier time embedding.
5353
flip_sin_to_cos (`bool`, *optional*, defaults to :
54-
obj:`False`): Whether to flip sin to cos for fourier time embedding.
54+
obj:`True`): Whether to flip sin to cos for fourier time embedding.
5555
down_block_types (`Tuple[str]`, *optional*, defaults to :
5656
obj:`("DownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D")`): Tuple of downsample block
5757
types.

src/diffusers/models/unet_2d_condition.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ class UNet2DConditionModel(ModelMixin, ConfigMixin):
6060
in_channels (`int`, *optional*, defaults to 4): The number of channels in the input sample.
6161
out_channels (`int`, *optional*, defaults to 4): The number of channels in the output.
6262
center_input_sample (`bool`, *optional*, defaults to `False`): Whether to center the input sample.
63-
flip_sin_to_cos (`bool`, *optional*, defaults to `False`):
63+
flip_sin_to_cos (`bool`, *optional*, defaults to `True`):
6464
Whether to flip the sin to cos in the time embedding.
6565
freq_shift (`int`, *optional*, defaults to 0): The frequency shift to apply to the time embedding.
6666
down_block_types (`Tuple[str]`, *optional*, defaults to `("CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "DownBlock2D")`):

0 commit comments

Comments
 (0)