model_config.json

{
    "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"
}