31 lines
879 B
JSON
31 lines
879 B
JSON
{
|
|
"output_dir": "lora_models/xander_sd15_final",
|
|
"sd_model_version": "sd15",
|
|
"lora_training_urls": "https://storage.googleapis.com/public-assets-xander/A_workbox/lora_training_sets/xander_big.zip",
|
|
"concept_mode": "face",
|
|
"seed": 0,
|
|
"resolution": 512,
|
|
"train_batch_size": 4,
|
|
"n_sample_imgs": 6,
|
|
"max_train_steps": 600,
|
|
"token_warmup_steps": 0,
|
|
"checkpointing_steps": 100,
|
|
"gradient_accumulation_steps": 1,
|
|
|
|
"sample_imgs_lora_scale": 0.8,
|
|
"n_tokens": 2,
|
|
"ti_lr": 0.001,
|
|
"remove_ti_token_from_prompts": false,
|
|
|
|
"text_encoder_lora_optimizer": null,
|
|
"text_encoder_lora_lr": 0.5e-4,
|
|
"text_encoder_lora_weight_decay": 1e-5,
|
|
"text_encoder_lora_rank": 16,
|
|
|
|
"unet_lr": 0.001,
|
|
"lora_alpha_multiplier": 1.0,
|
|
"lora_rank": 16,
|
|
"use_dora": false,
|
|
"caption_model": "gpt4-v",
|
|
"debug": true
|
|
} |