Added support to create asymmetrical U-Net structures #5400
patrickvonplaten
merged
36 commits into
huggingface:main
from
main
Added args, kwargs to ```U
b8dca9f0
Add UNetMidBlock2D as a supported mid block type
577d1e24
Fix extra init input for UNetMidBlock2D, change allowed types for Mid…
2c8d8043
Update unet_2d_condition.py
e07c6156
Update unet_2d_condition.py
8344b2d2
Update unet_2d_condition.py
025b1ec2
Update unet_2d_condition.py
62302148
Update unet_2d_condition.py
80b891b6
Update unet_2d_condition.py
c4e4d40d
Update unet_2d_condition.py
6176fd5a
Update unet_2d_condition.py
73737f8c
Update unet_2d_blocks.py
bc53a32e
Update unet_2d_blocks.py
314730ba
Update unet_2d_blocks.py
545a0b0f
Update unet_2d_condition.py
bf772686
Update unet_2d_blocks.py
1998c17f
sayakpaul
commented
on 2023-10-16
Updated docstring, increased check strictness
81d682ed
patrickvonplaten
commented
on 2023-10-16
patrickvonplaten
approved these changes
on 2023-10-16
Add basic shape-check test for asymmetrical unets
93ce7d3d
sayakpaul
commented
on 2023-10-17
sayakpaul
commented
on 2023-10-17
sayakpaul
commented
on 2023-10-17
sayakpaul
commented
on 2023-10-17
sayakpaul
approved these changes
on 2023-10-17
Update src/diffusers/models/unet_2d_blocks.py
e6b937cc
Merge branch 'main' into main
3cf5e20f
Update unet_2d_condition.py
0e9f6f48
Update unet_2d_condition.py
a91d0856
Fixed docstring and wrong default value
d75f2039
Merge branch 'main' into main
4cf629bc
Reformat with black
97c47b60
Merge branch 'main' into main
76955663
Merge branch 'main' into main
b3bf008d
Reformat with necessary commands
196ab3e7
Merge branch 'huggingface:main' into main
a65ec270
Add UNetMidBlockFlat to versatile_diffusion/modeling_text_unet.py to …
bbebc237
sayakpaul
approved these changes
on 2023-10-18
patrickvonplaten
commented
on 2023-10-18
patrickvonplaten
commented
on 2023-10-18
Merge branch 'main' into main
b6c0ff67
Removed args, kwargs, use on mid-block type
0a73108a
Merge remote-tracking branch 'refs/remotes/origin/main'
950d9695
Make fix-copies
bfbf85c3
sayakpaul
commented
on 2023-10-20
sayakpaul
approved these changes
on 2023-10-20
Update src/diffusers/models/unet_2d_condition.py
8389a851
make fix-copies
252ca850
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub