train_lora/
train-lora.rs

1use stable_diffusion_trainer::*;
2
3fn main() {
4    let kohya_ss = std::env::var("KOHYA_SS_PATH").expect("KOHYA_SS_PATH not set");
5    let environment = Environment::new().with_kohya_ss(kohya_ss);
6
7    let prompt = Prompt::new("bacana", "white dog");
8    let image_data_set = ImageDataSet::from_dir("examples/training/lora/bacana/images");
9    let data_set = TrainingDataSet::new(image_data_set);
10    let output = Output::new("{prompt.instance}({prompt.class})d{network.dimension}a{network.alpha}", "examples/training/lora/bacana/output");
11    let parameters = Parameters::new(prompt, data_set, output);
12
13    Trainer::new()
14        .with_environment(environment)
15        .start(&parameters);
16}
train_lora/train-lora.rs

train_lora/
train-lora.rs
