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 b1e2f3a commit f94b442Copy full SHA for f94b442
models/unet_2d_blocks.py
@@ -2211,7 +2211,7 @@ def __init__(
2211
dim_head=attn_num_head_channels if attn_num_head_channels is not None else out_channels,
2212
rescale_output_factor=output_scale_factor,
2213
eps=resnet_eps,
2214
- norm_num_groups=resnet_groups if resnet_time_scale_shift == "default" else None,
+ norm_num_groups=resnet_groups if resnet_time_scale_shift != "spatial" else None,
2215
spatial_norm_dim=temb_channels if resnet_time_scale_shift == "spatial" else None,
2216
residual_connection=True,
2217
bias=True,
0 commit comments