Struct Flux2VaeConfig 

pub struct Flux2VaeConfig {
    pub in_channels: usize,
    pub out_channels: usize,
    pub latent_channels: usize,
    pub layers_per_block: usize,
    pub norm_num_groups: usize,
    pub block_out_channels: Vec<usize>,
    pub act_fn: String,
    pub batch_norm_eps: f32,
    pub mid_block_add_attention: bool,
    pub use_post_quant_conv: bool,
    pub scaling_factor: f32,
    pub shift_factor: f32,
}

Fields

in_channels: usize

out_channels: usize

latent_channels: usize

layers_per_block: usize

norm_num_groups: usize

block_out_channels: Vec<usize>

act_fn: String

batch_norm_eps: f32

mid_block_add_attention: bool

use_post_quant_conv: bool

scaling_factor: f32

shift_factor: f32

Implementations

impl Flux2VaeConfig

pub fn from_file(path: &Path) -> Result<Flux2VaeConfig, Error>

pub fn flux2_klein() -> Flux2VaeConfig

pub fn tiny() -> Flux2VaeConfig

Tiny VAE for unit tests (no mid attention to shrink graph).

pub fn bn_channels(&self) -> usize

pub fn encode_spatial_stride(&self) -> usize

Spatial downsample factor from RGB to latent (one halving per encoder down block except the last).

Trait Implementations

impl Clone for Flux2VaeConfig

fn clone(&self) -> Flux2VaeConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Flux2VaeConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Flux2VaeConfig

fn deserialize<__D>( __deserializer: __D, ) -> Result<Flux2VaeConfig, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.