[Big refactor] move unets to `unets` module 🦋 #6630
move unets to module 🦋
9533ee15
sayakpaul
changed the title [Refactor] move unets to `unets` module 🦋 [Big refactor] move unets to `unets` module 🦋 2 years ago
parameterize unet-level import.
492b9b96
fix flax unet2dcondition model import
9bec7753
models __init__
9496d72a
Merge branch 'main' into unet-module
52f76ae0
mildly depcrecating models.unet_2d_blocks in favor of models.unets.un…
85c20aa0
noqa
5dc5886f
correct depcrecation behaviour
233ac980
inherit from the actual classes.
4e605911
Empty-Commit
e2033611
backwards compatibility for unet_2d.py
a7adbeaa
backward compatibility for unet_2d_condition
c8b433f5
bc for unet_1d
5b0590f1
bc for unet_1d_blocks
37f965e7
Merge branch 'main' into unet-module
023545bd
merge main and resolve conflicts
ebdffb2e
Merge branch 'main' into unet-module
c110003e
DN6
approved these changes
on 2024-01-22
Merge branch 'main' into unet-module
4c4e7fbc
sayakpaul
merged
1f0705ad
into main 2 years ago
sayakpaul
deleted the unet-module branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub