Module candle_transformers::models::stable_diffusion::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.