39 lines
1.2 KiB
JSON
39 lines
1.2 KiB
JSON
{
|
|
"output_dir": "lora_models/banny",
|
|
"sd_model_version": "sdxl",
|
|
"lora_training_urls": "https://storage.googleapis.com/public-assets-xander/A_workbox/lora_training_sets/banny_best.zip",
|
|
"concept_mode": "object",
|
|
"seed": 1,
|
|
"resolution": 512,
|
|
"validation_img_size": [1024, 1024],
|
|
"train_batch_size": 4,
|
|
"n_sample_imgs": 6,
|
|
"max_train_steps": 320,
|
|
"txt_encoders_lr_warmup_steps": 60,
|
|
"checkpointing_steps": 80,
|
|
"gradient_accumulation_steps": 1,
|
|
"n_tokens": 2,
|
|
"ti_lr": 0.001,
|
|
"ti_weight_decay": 0.0002,
|
|
"text_encoder_lora_optimizer": "adamw",
|
|
"text_encoder_lora_lr": 0.5e-4,
|
|
"text_encoder_lora_weight_decay": 1e-5,
|
|
"text_encoder_lora_rank": 12,
|
|
"off_ratio_power": 0.05,
|
|
"prodigy_d_coef": 0.7,
|
|
"lora_weight_decay": 0.001,
|
|
"l1_penalty": 0.1,
|
|
"noise_offset": 0.05,
|
|
"snr_gamma": 5.0,
|
|
"lora_alpha_multiplier": 1.0,
|
|
"lora_rank": 16,
|
|
"use_dora": false,
|
|
"caption_model": "blip",
|
|
"left_right_flip_augmentation": true,
|
|
"aspect_ratio_bucketing": false,
|
|
"augment_imgs_up_to_n": 20,
|
|
"verbose": true,
|
|
"debug": true,
|
|
"weight_type": "bf16",
|
|
"dataloader_num_workers": 0
|
|
} |