Struct AutoEncoderKL 

pub struct AutoEncoderKL { /* private fields */ }

Z-Image VAE (AutoEncoderKL) - Diffusers Format

Implementations

impl AutoEncoderKL

pub fn new(cfg: &VaeConfig, vb: VarBuilder<'_>) -> Result<Self>

pub fn encode(&self, xs: &Tensor) -> Result<Tensor>

Encode image to latent space xs: (B, 3, H, W) RGB image, range [-1, 1] Returns: (B, latent_channels, H/8, W/8)

pub fn decode(&self, xs: &Tensor) -> Result<Tensor>

Decode latent to image xs: (B, latent_channels, H/8, W/8) Returns: (B, 3, H, W) RGB image, range [-1, 1]

pub fn scale_factor(&self) -> f64

Get scaling factor

pub fn shift_factor(&self) -> f64

Get shift factor

Trait Implementations

impl Clone for AutoEncoderKL

fn clone(&self) -> AutoEncoderKL

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for AutoEncoderKL

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

Formats the value using the given formatter.

impl Module for AutoEncoderKL

fn forward(&self, xs: &Tensor) -> Result<Tensor>