| { | |
| "pretrained_model_name_or_path": "./output/checkpoint_last", | |
| "pretrained_vae_name_or_path": null, | |
| "tokenizer_name": null, | |
| "instance_data_dir": "./instance-images", | |
| "class_data_dir": "./class-images", | |
| "instance_prompt": "best quality, masterpiece, highres, Loliwm 1girl, Beautiful face", | |
| "class_prompt": "best quality, masterpiece, highres, 1girl, Beautiful face", | |
| "class_negative_prompt": "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry", | |
| "save_sample_prompt": "best quality, masterpiece, highres, Loliwm 1girl, Beautiful face, looking at viewer", | |
| "save_sample_negative_prompt": "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry", | |
| "n_save_sample": 4, | |
| "save_guidance_scale": 7.5, | |
| "save_infer_steps": 50, | |
| "with_prior_preservation": true, | |
| "pad_tokens": false, | |
| "prior_loss_weight": 1.0, | |
| "num_class_images": 9295, | |
| "output_dir": "./output", | |
| "seed": 1337, | |
| "resolution": 512, | |
| "center_crop": false, | |
| "train_text_encoder": false, | |
| "train_batch_size": 1, | |
| "sample_batch_size": 4, | |
| "num_train_epochs": 1, | |
| "max_train_steps": 3000, | |
| "gradient_accumulation_steps": 1, | |
| "gradient_checkpointing": true, | |
| "learning_rate": 5e-06, | |
| "scale_lr": false, | |
| "scale_lr_sqrt": false, | |
| "lr_scheduler": "cosine_with_restarts", | |
| "lr_warmup_steps": 100, | |
| "use_8bit_adam": true, | |
| "use_deepspeed_adam": false, | |
| "optimizer": "adamw", | |
| "adam_beta1": 0.9, | |
| "adam_beta2": 0.999, | |
| "adam_epsilon": 1e-08, | |
| "sgd_momentum": 0.9, | |
| "sgd_dampening": 0, | |
| "max_grad_norm": 1.0, | |
| "weight_decay": 0.01, | |
| "logging_dir": "logs", | |
| "log_interval": 10, | |
| "save_interval": 500, | |
| "save_min_steps": 10, | |
| "mixed_precision": "fp16", | |
| "not_cache_latents": false, | |
| "local_rank": -1, | |
| "concepts_list": [ | |
| { | |
| "instance_prompt": "best quality, masterpiece, highres, Loliwm 1girl, Beautiful face", | |
| "class_prompt": "best quality, masterpiece, highres, 1girl, Beautiful face", | |
| "class_negative_prompt": "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry", | |
| "instance_data_dir": "./instance-images", | |
| "class_data_dir": "./class-images" | |
| } | |
| ], | |
| "wandb": false, | |
| "wandb_artifact": false, | |
| "rm_after_wandb_saved": false, | |
| "wandb_name": "Stable-Diffusion-Dreambooth", | |
| "read_prompt_filename": false, | |
| "read_prompt_txt": false, | |
| "append_prompt": "instance", | |
| "save_unet_half": false, | |
| "unet_half": false, | |
| "clip_skip": 1, | |
| "num_cycles": 1, | |
| "last_epoch": -1, | |
| "use_aspect_ratio_bucket": false, | |
| "debug_arb": false, | |
| "save_optimizer": true, | |
| "save_states": true, | |
| "resume": false, | |
| "resume_from": "output/checkpoint_last", | |
| "config": null, | |
| "arb_dim_limit": 1024, | |
| "arb_divisible": 64, | |
| "arb_max_ar_error": 4, | |
| "arb_max_size": [ | |
| 768, | |
| 512 | |
| ], | |
| "arb_min_dim": 256, | |
| "deepdanbooru": false, | |
| "dd_threshold": 0.6, | |
| "dd_alpha_sort": false, | |
| "dd_use_spaces": true, | |
| "dd_use_escape": true, | |
| "enable_rotate": false, | |
| "dd_include_ranks": false, | |
| "use_ema": false, | |
| "ucg": 0.0, | |
| "debug_prompt": false, | |
| "xformers": false | |
| } |