Expand description
2D UNet Building Blocks
Structsยง
- Cross
Attn Down Block2D - Cross
Attn Down Block2D Config - Cross
Attn UpBlock2D - Cross
Attn UpBlock2D Config - Down
Block2D - Down
Block2D Config - Down
Encoder Block2D - Down
Encoder Block2D Config - UNet
MidBlock2D - UNet
MidBlock2D Config - UNet
MidBlock2D Cross Attn - UNet
MidBlock2D Cross Attn Config - UpBlock2D
- UpBlock2D
Config - UpDecoder
Block2D - UpDecoder
Block2D Config