Method load_model

cake-core/src/models/sd/vae.rs:88–106

(
        name: Option<String>,
        version: StableDiffusionVersion,
        use_f16: bool,
        device: &Device,
        dtype: DType,
        cache_dir: String,
        config: &StableDiffusi

impl VAE {
    pub fn load_model(
        name: Option<String>,
        version: StableDiffusionVersion,
        use_f16: bool,
        device: &Device,
        dtype: DType,
        cache_dir: String,
        config: &StableDiffusionConfig,
    ) -> anyhow::Result<Box<Self>>
    where
        Self: Sized,
    {
        let vae_weights = ModelFile::Vae.get(name, version, use_f16, cache_dir)?;
        let vae_model = config.build_vae(vae_weights, device, dtype)?;

        info!("Loading VAE model...");

        Ok(Box::new(Self { vae_model }))
    }

    pub async fn encode(
        forwarder: &mut Box<dyn Forwarder>,