Struct Environment 

pub struct Environment { /* private fields */ }

The environment structure.

Implementations

impl Environment

pub fn load() -> Self

Load the environment from the configuration file.

pub fn save(&self) -> Result<(), Box<dyn Error>>

Save the environment to the configuration file.

pub fn new() -> Self

Create a new environment structure.

Examples found in repository

examples/train-lora.rs (line 5)

fn main() {
    let kohya_ss = std::env::var("KOHYA_SS_PATH").expect("KOHYA_SS_PATH not set");
    let environment = Environment::new().with_kohya_ss(kohya_ss);

    let prompt = Prompt::new("bacana", "white dog");
    let image_data_set = ImageDataSet::from_dir("examples/training/lora/bacana/images");
    let data_set = TrainingDataSet::new(image_data_set);
    let output = Output::new("{prompt.instance}({prompt.class})d{network.dimension}a{network.alpha}", "examples/training/lora/bacana/output");
    let parameters = Parameters::new(prompt, data_set, output);

    Trainer::new()
        .with_environment(environment)
        .start(&parameters);
}

pub fn with_kohya_ss(self, kohya_ss: impl Into<PathBuf>) -> Self

Set the kohya_ss path.

Examples found in repository

examples/train-lora.rs (line 5)

fn main() {
    let kohya_ss = std::env::var("KOHYA_SS_PATH").expect("KOHYA_SS_PATH not set");
    let environment = Environment::new().with_kohya_ss(kohya_ss);

    let prompt = Prompt::new("bacana", "white dog");
    let image_data_set = ImageDataSet::from_dir("examples/training/lora/bacana/images");
    let data_set = TrainingDataSet::new(image_data_set);
    let output = Output::new("{prompt.instance}({prompt.class})d{network.dimension}a{network.alpha}", "examples/training/lora/bacana/output");
    let parameters = Parameters::new(prompt, data_set, output);

    Trainer::new()
        .with_environment(environment)
        .start(&parameters);
}

pub fn kohya_ss(&self) -> &Path

Get the kohya_ss path.

pub fn binary_path(&self) -> PathBuf

Get the kohya_ss path.

pub fn python_executable_path(&self) -> PathBuf

Get the kohya_ss path.

pub fn activate(&mut self)

Activate the environment.

pub fn deactivate(&mut self)

Deactivate the environment.

Trait Implementations

impl Debug for Environment

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

Formats the value using the given formatter.

impl Default for Environment

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for Environment

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for Environment

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.