text-to-image

training params

{
    "pretrained_model_name_or_path": "CompVis/stable-diffusion-v1-4",
    "instance_data_dir": "./a9054d36-59d1-4374-ab1f-2ca457b539e2/instance_data",
    "class_data_dir": "./class_data/a-portrait-of-a-person",
    "output_dir": "./a9054d36-59d1-4374-ab1f-2ca457b539e2/",
    "with_prior_preservation": true,
    "prior_loss_weight": 1.0,
    "instance_prompt": "a portrait of [V]",
    "class_prompt": "a portrait of a person",
    "resolution": 512,
    "train_batch_size": 1,
    "gradient_accumulation_steps": 1,
    "gradient_checkpointing": true,
    "use_8bit_adam": true,
    "learning_rate": 5e-06,
    "lr_scheduler": "constant",
    "lr_warmup_steps": 0,
    "num_class_images": 200,
    "max_train_steps": 1050,
    "mixed_precision": "fp16"
}