Struct ModelConfigBuilder

pub struct ModelConfigBuilder { /* private fields */ }

Builder for ModelConfig.

Implementations

impl ModelConfigBuilder

pub fn upscale_model<VALUE: Into<CLibPath>>( &mut self, value: VALUE, ) -> &mut Self

Path to esrgan model. Upscale images after generate, just RealESRGAN_x4plus_anime_6B supported by now

pub fn upscale_repeats<VALUE: Into<i32>>(&mut self, value: VALUE) -> &mut Self

Run the ESRGAN upscaler this many times (default 1)

pub fn model<VALUE: Into<CLibPath>>(&mut self, value: VALUE) -> &mut Self

Path to full model

pub fn diffusion_model<VALUE: Into<CLibPath>>( &mut self, value: VALUE, ) -> &mut Self

Path to the standalone diffusion model

pub fn clip_l<VALUE: Into<CLibPath>>(&mut self, value: VALUE) -> &mut Self

path to the clip-l text encoder

pub fn clip_g<VALUE: Into<CLibPath>>(&mut self, value: VALUE) -> &mut Self

path to the clip-g text encoder

pub fn t5xxl<VALUE: Into<CLibPath>>(&mut self, value: VALUE) -> &mut Self

Path to the t5xxl text encoder

pub fn vae<VALUE: Into<CLibPath>>(&mut self, value: VALUE) -> &mut Self

Path to vae

pub fn taesd<VALUE: Into<CLibPath>>(&mut self, value: VALUE) -> &mut Self

Path to taesd. Using Tiny AutoEncoder for fast decoding (low quality)

pub fn control_net<VALUE: Into<CLibPath>>(&mut self, value: VALUE) -> &mut Self

Path to control net model

pub fn embeddings<VALUE: Into<CLibPath>>(&mut self, value: VALUE) -> &mut Self

Path to embeddings

pub fn stacked_id_embd<VALUE: Into<CLibPath>>( &mut self, value: VALUE, ) -> &mut Self

Path to PHOTOMAKER stacked id embeddings

pub fn weight_type<VALUE: Into<WeightType>>( &mut self, value: VALUE, ) -> &mut Self

Weight type. If not specified, the default is the type of the weight file

pub fn vae_tiling<VALUE: Into<bool>>(&mut self, value: VALUE) -> &mut Self

Process vae in tiles to reduce memory usage (default: false)

pub fn rng<VALUE: Into<RngFunction>>(&mut self, value: VALUE) -> &mut Self

RNG (default: CUDA)

pub fn schedule<VALUE: Into<Schedule>>(&mut self, value: VALUE) -> &mut Self

Denoiser sigma schedule (default: DEFAULT)

pub fn vae_on_cpu<VALUE: Into<bool>>(&mut self, value: VALUE) -> &mut Self

Keep vae in cpu (for low vram) (default: false)

pub fn clip_on_cpu<VALUE: Into<bool>>(&mut self, value: VALUE) -> &mut Self

keep clip in cpu (for low vram) (default: false)

pub fn control_net_cpu<VALUE: Into<bool>>(&mut self, value: VALUE) -> &mut Self

Keep controlnet in cpu (for low vram) (default: false)

pub fn flash_attention<VALUE: Into<bool>>(&mut self, value: VALUE) -> &mut Self

Use flash attention in the diffusion model (for low vram). Might lower quality, since it implies converting k and v to f16. This might crash if it is not supported by the backend.

pub fn chroma_use_dit_mask<VALUE: Into<bool>>( &mut self, value: VALUE, ) -> &mut Self

pub fn chroma_use_t5_mask<VALUE: Into<bool>>( &mut self, value: VALUE, ) -> &mut Self

pub fn chroma_t5_mask_pad<VALUE: Into<i32>>( &mut self, value: VALUE, ) -> &mut Self

pub fn build(&self) -> Result<ModelConfig, ConfigBuilderError>

Builds a new ModelConfig.

Errors

If a required field has not been initialized.

impl ModelConfigBuilder

pub fn lora_model(&mut self, lora_model: &Path) -> &mut Self

pub fn n_threads(&mut self, value: i32) -> &mut Self

Trait Implementations

impl Clone for ModelConfigBuilder

fn clone(&self) -> ModelConfigBuilder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Default for ModelConfigBuilder

fn default() -> Self

Returns the “default value” for a type.