File size: 1,303 Bytes
7a9359c bf4f8d4 7a9359c bf4f8d4 7a9359c bf4f8d4 7a9359c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
{
"pretrained_model_name_or_path": "stabilityai/stable-diffusion-2-1-base",
"revision": "fp16",
"instance_data_dir": "images/mushrooms",
"class_data_dir": "images/doy pack",
"instance_prompt": "xyz doy pack",
"class_prompt": "a doy pack",
"save_sample_prompt": null,
"save_sample_negative_prompt": null,
"n_save_sample": 5,
"save_guidance_scale": 7.5,
"save_infer_steps": 50,
"pad_tokens": false,
"prior_loss_weight": 1.0,
"num_class_images": 300,
"output_dir": "results",
"seed": 1234,
"resolution": 512,
"center_crop": false,
"train_batch_size": 1,
"sample_batch_size": 5,
"max_train_steps": 3000,
"max_train_text_encoder": 1500,
"gradient_checkpointing": false,
"learning_rate": 1e-06,
"lr_scheduler": "constant",
"lr_warmup_steps": 500,
"adam_beta1": 0.9,
"adam_beta2": 0.999,
"adam_weight_decay": 0.01,
"adam_epsilon": 1e-08,
"max_grad_norm": 1.0,
"log_interval": 10,
"save_interval": 10000,
"save_min_steps": 5000,
"mixed_precision": "fp16",
"concepts_list": [
{
"instance_prompt": "xyz doy pack",
"class_prompt": "a doy pack",
"instance_data_dir": "images/mushrooms",
"class_data_dir": "images/doy pack"
}
],
"repo_name": "mush",
"instance_class": "doy pack",
"num_train_epochs": 10
} |