diffusers
Fix Consistency Models UNet2DMidBlock2D Attention GroupNorm Bug
#4863
Merged
Merged
Fix Consistency Models UNet2DMidBlock2D Attention GroupNorm Bug #4863
sayakpaul
merged
7 commits into
huggingface:main
from
dg845:consistency-models-fix-mid-block
Add attn_groups argument to UNet2DMidBlock2D to control theinternal A…
ad970cec
sayakpaul
commented
on 2023-09-01
Add docstring for attn_norm_num_groups in UNet2DModel.
347fc547
Since the test UNet config uses resnet_time_scale_shift == 'scale_shi…
eee35092
patrickvonplaten
commented
on 2023-09-01
Add test for attn_norm_num_groups to UNet2DModelTests.
b7b29238
Fix expected slices for slow tests.
b99373ba
Also fix tolerances for slow tests.
2884fa1b
patrickvonplaten
approved these changes
on 2023-09-14
Merge branch 'main' into consistency-models-fix-mid-block
c4913011
sayakpaul
approved these changes
on 2023-09-15
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub