Struct StableDiffusionConfig 

pub struct StableDiffusionConfig {
    pub width: i64,
    pub height: i64,
    pub clip: Config,
    /* private fields */
}

Fields

width: i64

height: i64

clip: Config

Implementations

impl StableDiffusionConfig

pub fn v1_5( sliced_attention_size: Option<i64>, height: Option<i64>, width: Option<i64>, ) -> Self

pub fn v2_1( sliced_attention_size: Option<i64>, height: Option<i64>, width: Option<i64>, ) -> Self

pub fn v2_1_inpaint( sliced_attention_size: Option<i64>, height: Option<i64>, width: Option<i64>, ) -> Self

pub fn build_vae( &self, vae_weights: &str, device: Device, ) -> Result<AutoEncoderKL>

pub fn build_unet( &self, unet_weights: &str, device: Device, in_channels: i64, ) -> Result<UNet2DConditionModel>

pub fn build_scheduler(&self, n_steps: usize) -> DDIMScheduler

pub fn build_clip_transformer( &self, clip_weights: &str, device: Device, ) -> Result<ClipTextTransformer>

Trait Implementations

impl Clone for StableDiffusionConfig

fn clone(&self) -> StableDiffusionConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StableDiffusionConfig

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

Formats the value using the given formatter.