candle_transformers
0.5.0
Module unet_2d_blocks
Structs
In candle_transformers::models::stable_diffusion
?
Settings
Module
candle_transformers
::
models
::
stable_diffusion
::
unet_2d_blocks
Copy item path
source
·
[
−
]
Expand description
2D UNet Building Blocks
Structs
§
CrossAttnDownBlock2D
CrossAttnDownBlock2DConfig
CrossAttnUpBlock2D
CrossAttnUpBlock2DConfig
DownBlock2D
DownBlock2DConfig
DownEncoderBlock2D
DownEncoderBlock2DConfig
UNetMidBlock2D
UNetMidBlock2DConfig
UNetMidBlock2DCrossAttn
UNetMidBlock2DCrossAttnConfig
UpBlock2D
UpBlock2DConfig
UpDecoderBlock2D
UpDecoderBlock2DConfig