mirror of https://github.com/bmaltais/kohya_ss
42 lines
1.1 KiB
JSON
42 lines
1.1 KiB
JSON
{
|
|
"general": {
|
|
"shuffle_caption": true,
|
|
"caption_extension": ".txt",
|
|
"keep_tokens": 1,
|
|
"seed": 47
|
|
},
|
|
"model": {
|
|
"pretrained_model_name_or_path": "stabilityai/stable-diffusion-xl-base-1.0",
|
|
"vae": "stabilityai/sd-vae-ft-mse"
|
|
},
|
|
"training": {
|
|
"resolution": "768,768",
|
|
"batch_size": 1,
|
|
"learning_rate": 0.00015,
|
|
"lr_scheduler": "cosine_with_restarts",
|
|
"max_train_steps": 4000,
|
|
"optimizer": "adamw8bit",
|
|
"mixed_precision": "fp16",
|
|
"gradient_checkpointing": true,
|
|
"clip_skip": 2,
|
|
"network_dim": 32,
|
|
"network_alpha": 16,
|
|
"save_precision": "fp16",
|
|
"save_every_n_steps": 1000
|
|
},
|
|
"folders": {
|
|
"train_data_dir": "./data/train",
|
|
"reg_data_dir": "./data/reg",
|
|
"output_dir": "./output_5080",
|
|
"logging_dir": "./logs_5080"
|
|
},
|
|
"advanced": {
|
|
"bucket_reso_steps": 64,
|
|
"bucket_no_upscale": true,
|
|
"xformers": true,
|
|
"cache_latents": true,
|
|
"min_bucket_reso": 320,
|
|
"max_bucket_reso": 768
|
|
}
|
|
}
|
|
|