Skip to content

GitLab

Unverified Commit df8559a7 authored by fpgaminer's avatar fpgaminer Committed by GitHub
Browse files

Fix: UNet2DModel::__init__ type hints; fixes issue #4806 (#7175) 


Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 8f206a58
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/diffusers/models/unets/unet_2d.py
View file @ df8559a7
...@@ -99,9 +99,9 @@ class UNet2DModel(ModelMixin, ConfigMixin): ...@@ -99,9 +99,9 @@ class UNet2DModel(ModelMixin, ConfigMixin):
time_embedding_type: str = "positional", time_embedding_type: str = "positional",
freq_shift: int = 0, freq_shift: int = 0,
flip_sin_to_cos: bool = True, flip_sin_to_cos: bool = True,
down_block_types: Tuple[str] = ("DownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D"), down_block_types: Tuple[str, ...] = ("DownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D"),
up_block_types: Tuple[str] = ("AttnUpBlock2D", "AttnUpBlock2D", "AttnUpBlock2D", "UpBlock2D"), up_block_types: Tuple[str, ...] = ("AttnUpBlock2D", "AttnUpBlock2D", "AttnUpBlock2D", "UpBlock2D"),
block_out_channels: Tuple[int] = (224, 448, 672, 896), block_out_channels: Tuple[int, ...] = (224, 448, 672, 896),
layers_per_block: int = 2, layers_per_block: int = 2,
mid_block_scale_factor: float = 1, mid_block_scale_factor: float = 1,
downsample_padding: int = 1, downsample_padding: int = 1,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment