Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
e5215dee
Unverified
Commit
e5215dee
authored
May 25, 2023
by
YiYi Xu
Committed by
GitHub
May 25, 2023
Browse files
fix broken change for vq pipeline (#3563)
fix vq_model Co-authored-by:
yiyixuxu
<
yixu310@gmail,com
>
parent
03b7a84c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/diffusers/models/unet_2d_blocks.py
src/diffusers/models/unet_2d_blocks.py
+1
-1
No files found.
src/diffusers/models/unet_2d_blocks.py
View file @
e5215dee
...
...
@@ -2211,7 +2211,7 @@ class AttnUpDecoderBlock2D(nn.Module):
dim_head
=
attn_num_head_channels
if
attn_num_head_channels
is
not
None
else
out_channels
,
rescale_output_factor
=
output_scale_factor
,
eps
=
resnet_eps
,
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
,
spatial_norm_dim
=
temb_channels
if
resnet_time_scale_shift
==
"spatial"
else
None
,
residual_connection
=
True
,
bias
=
True
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment