sd_dreambooth_extension/templates/defaults/dreambooth_model_config.json

90 lines
2.4 KiB
JSON

{
"weight_decay": 0.01,
"cache_latents": true,
"clip_skip": 2,
"concepts_list": [],
"concepts_path": "",
"custom_model_name": "",
"deterministic": false,
"disable_class_matching": false,
"disable_logging": false,
"ema_predict": false,
"epoch_pause_frequency": 0,
"epoch_pause_time": 0,
"freeze_clip_normalization": true,
"gradient_accumulation_steps": 1,
"gradient_checkpointing": true,
"gradient_set_to_none": true,
"graph_smoothing": 50,
"half_model": false,
"hflip": false,
"infer_ema": false,
"learning_rate": 2e-06,
"learning_rate_min": 1e-06,
"lora_learning_rate": 0.0001,
"lora_txt_learning_rate": 5e-05,
"lora_txt_rank": 4,
"lora_txt_weight": 1.0,
"lora_unet_rank": 4,
"lora_weight": 1.0,
"lora_use_buggy_requires_grad": false,
"lr_cycles": 1,
"lr_factor": 0.5,
"lr_power": 1.0,
"lr_scale_pos": 0.5,
"lr_scheduler": "constant_with_warmup",
"lr_warmup_steps": 500,
"max_token_length": 75,
"mixed_precision": "fp16",
"noise_scheduler": "DDPM",
"num_train_epochs": 200,
"offset_noise": 0,
"optimizer": "8bit AdamW",
"pad_tokens": true,
"pretrained_vae_name_or_path": "",
"prior_loss_scale": false,
"prior_loss_target": 100,
"prior_loss_weight": 0.75,
"prior_loss_weight_min": 0.1,
"resolution": 512,
"revision": 21910,
"sample_batch_size": 1,
"sanity_prompt": "",
"sanity_seed": 420420,
"save_ckpt_after": true,
"save_ckpt_cancel": false,
"save_ckpt_during": true,
"save_ema": true,
"save_embedding_every": 25,
"save_lora_after": true,
"save_lora_cancel": false,
"save_lora_during": true,
"save_lora_for_extra_net": true,
"save_preview_every": 5,
"save_safetensors": true,
"save_state_after": false,
"save_state_cancel": false,
"save_state_during": false,
"scheduler": "UniPCMultistep",
"shuffle_tags": true,
"split_loss": true,
"stop_text_encoder": 0.75,
"strict_tokens": true,
"dynamic_img_norm": false,
"tenc_weight_decay": 0.01,
"tenc_grad_clip_norm": 6,
"tomesd": 0,
"train_batch_size": 1,
"train_imagic": false,
"train_unet": true,
"train_unfrozen": true,
"txt_learning_rate": 1e-06,
"use_concepts": true,
"use_ema": false,
"use_lora": false,
"use_lora_extended": false,
"use_shared_src": [
false
],
"use_subdir": false
}