Struct StableDiffusionConfig

pub struct StableDiffusionConfig {
    pub width: usize,
    pub height: usize,
    pub clip: Config,
    pub clip2: Option<Config>,
    /* private fields */
}

Fields

width: usize

height: usize

clip: Config

clip2: Option<Config>

Implementations

impl StableDiffusionConfig

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

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

pub fn sdxl( sliced_attention_size: Option<usize>, height: Option<usize>, width: Option<usize>, ) -> Self

pub fn sdxl_turbo( sliced_attention_size: Option<usize>, height: Option<usize>, width: Option<usize>, ) -> Self

pub fn ssd1b( sliced_attention_size: Option<usize>, height: Option<usize>, width: Option<usize>, ) -> Self

pub fn build_vae<P: AsRef<Path>>( &self, vae_weights: P, device: &Device, dtype: DType, ) -> Result<AutoEncoderKL>

pub fn build_unet<P: AsRef<Path>>( &self, unet_weights: P, device: &Device, in_channels: usize, use_flash_attn: bool, dtype: DType, ) -> Result<UNet2DConditionModel>

pub fn build_scheduler(&self, n_steps: usize) -> Result<Box<dyn Scheduler>>

Trait Implementations

impl Clone for StableDiffusionConfig

fn clone(&self) -> StableDiffusionConfig

Returns a copy 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.