Module unet_2d

candle_transformers::models::stable_diffusion

Module unet_2d 

Source
Expand description

2D UNet Denoising Models

The 2D Unet models take as input a noisy sample and the current diffusion timestep and return a denoised version of the input.

Structsยง

BlockConfig
UNet2DConditionModel
UNet2DConditionModelConfig