| { |
| "pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5", |
| "revision": null, |
| "tokenizer_name": null, |
| "instance_data_dir": "data/FFHQ_last2000_512", |
| "class_data_dir": null, |
| "instance_prompt": "photo of a person", |
| "class_prompt": null, |
| "with_prior_preservation": false, |
| "with_prior_preservation_attn": false, |
| "with_prior_preservation_lpips": false, |
| "with_prior_preservation_unet": false, |
| "type_prior_preservation_unet": "SD-SD", |
| "with_prior_preservation_clip": false, |
| "type_prior_preservation_clip": "img-img", |
| "with_prior_preservation_sks": false, |
| "prior_preservation_sks_step": 50, |
| "prior_loss_weight": 1.0, |
| "prior_attn_loss_weight": 0, |
| "prior_lpips_loss_weight": 0, |
| "prior_unet_loss_weight": 0, |
| "prior_clip_loss_weight": 0, |
| "prior_sks_loss_weight": 0, |
| "num_class_images": 100, |
| "output_dir": "./saved_model/age/0427_evaluation/finetune_double_prompt_150_random", |
| "seed": null, |
| "resolution": 512, |
| "center_crop": false, |
| "train_text_encoder": false, |
| "train_batch_size": 1, |
| "sample_batch_size": 4, |
| "num_train_epochs": 1, |
| "max_train_steps": 150, |
| "checkpointing_steps": 500, |
| "resume_from_checkpoint": null, |
| "gradient_accumulation_steps": 1, |
| "gradient_checkpointing": false, |
| "learning_rate": 5e-06, |
| "scale_lr": false, |
| "lr_scheduler": "constant", |
| "lr_warmup_steps": 500, |
| "lr_num_cycles": 1, |
| "lr_power": 1.0, |
| "use_8bit_adam": false, |
| "adam_beta1": 0.9, |
| "adam_beta2": 0.999, |
| "adam_weight_decay": 0.01, |
| "adam_epsilon": 1e-08, |
| "max_grad_norm": 1.0, |
| "push_to_hub": false, |
| "hub_token": null, |
| "hub_model_id": null, |
| "logging_dir": "logs", |
| "allow_tf32": false, |
| "report_to": "tensorboard", |
| "mixed_precision": null, |
| "prior_generation_precision": null, |
| "local_rank": -1, |
| "enable_xformers_memory_efficient_attention": false, |
| "multi_prompt": false, |
| "replace_with_multi_prompt": false, |
| "instance_age_path": "data/FFHQ_all_ages.npy", |
| "filenames_to_use": "False" |
| } |
