Skip to content

Commit f94b442

Browse files
yiyixuxuyiyixuxu
and
yiyixuxu
authored
fix broken change for vq pipeline (huggingface#3563)
fix vq_model Co-authored-by: yiyixuxu <yixu310@gmail,com>
1 parent b1e2f3a commit f94b442

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/unet_2d_blocks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2211,7 +2211,7 @@ def __init__(
22112211
dim_head=attn_num_head_channels if attn_num_head_channels is not None else out_channels,
22122212
rescale_output_factor=output_scale_factor,
22132213
eps=resnet_eps,
2214-
norm_num_groups=resnet_groups if resnet_time_scale_shift == "default" else None,
2214+
norm_num_groups=resnet_groups if resnet_time_scale_shift != "spatial" else None,
22152215
spatial_norm_dim=temb_channels if resnet_time_scale_shift == "spatial" else None,
22162216
residual_connection=True,
22172217
bias=True,

0 commit comments

Comments
 (0)