Struct DiffusionConfig 

pub struct DiffusionConfig {
    pub image_size: usize,
    pub in_channels: usize,
    pub model_channels: usize,
    pub num_res_blocks: usize,
    pub channel_mult: Vec<usize>,
    pub num_heads: usize,
    pub num_timesteps: usize,
    pub beta_schedule: BetaSchedule,
}

Diffusion model configuration

Fields

image_size: usize

Image size

in_channels: usize

Number of channels

model_channels: usize

Model channels

num_res_blocks: usize

Number of residual blocks per resolution

channel_mult: Vec<usize>

Channel multipliers for each resolution

num_heads: usize

Number of attention heads

num_timesteps: usize

Number of diffusion timesteps

beta_schedule: BetaSchedule

Beta schedule type

Implementations

impl DiffusionConfig

pub fn stable_diffusion() -> Self

Stable Diffusion configuration

pub fn ddpm() -> Self

DDPM configuration (original paper)

pub fn tiny() -> Self

Small model for testing

Trait Implementations

impl Clone for DiffusionConfig

fn clone(&self) -> DiffusionConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DiffusionConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for DiffusionConfig

fn default() -> Self

Returns the “default value” for a type.