Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "da9a54288e2d4b332adf7feabc4c5c1f49c292da"
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
Show 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):
time_embedding_type: str = "positional",
freq_shift: int = 0,
flip_sin_to_cos: bool = True,
down_block_types: Tuple[str] = ("DownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D"),
up_block_types: Tuple[str] = ("AttnUpBlock2D", "AttnUpBlock2D", "AttnUpBlock2D", "UpBlock2D"),
block_out_channels: Tuple[int] = (224, 448, 672, 896),
down_block_types: Tuple[str, ...] = ("DownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D"),
up_block_types: Tuple[str, ...] = ("AttnUpBlock2D", "AttnUpBlock2D", "AttnUpBlock2D", "UpBlock2D"),
block_out_channels: Tuple[int, ...] = (224, 448, 672, 896),
layers_per_block: int = 2,
mid_block_scale_factor: float = 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