Files
edenartlab-sd-lora-trainer/training_args.json
T
2024-03-29 23:21:58 -07:00

45 lines
1.2 KiB
JSON

{
"output_dir": "lora_models/clipx_tiny_no_cog",
"name": "unnamed",
"lora_training_urls": "https://storage.googleapis.com/public-assets-xander/A_workbox/lora_training_sets/clipx_tiny.zip",
"concept_mode": "style",
"sd_model_version": "sdxl",
"seed": 0,
"resolution": 1024,
"train_batch_size": 4,
"max_train_steps": 600,
"checkpointing_steps": 100,
"gradient_accumulation_steps": 1,
"prodigy_d_coef": 1.0,
"ti_lr": 0.001,
"ti_weight_decay": 0.0003,
"lora_weight_decay": 0.00,
"l1_penalty": 0.0,
"snr_gamma": 5.0,
"lora_rank": 12,
"use_dora": true,
"caption_prefix": "in the style of TOK, ",
"caption_model": "blip",
"left_right_flip_augmentation": true,
"augment_imgs_up_to_n": 20,
"verbose": true,
"debug": true,
"hard_pivot": false,
"off_ratio_power": 0.1,
"mixed_precision": "bf16",
"n_tokens": 2,
"inserting_list_tokens": [
"<s0>",
"<s1>"
],
"token_dict": {
"TOK": "<s0><s1>"
},
"scale_lr": false,
"unet_learning_rate": 1.0,
"lr_scheduler": "constant",
"lr_warmup_steps": 50,
"lr_num_cycles": 1,
"lr_power": 1.0,
"dataloader_num_workers": 0
}