config.py

# defaults
config = {
    "project_name": "default",
    "sd_version": "2.1",
    "output_dir": "output",
    "logging_dir": "logs",
    "resume_ckpt": "latest",
    "dataset_conditioning_folder": "root/conditioning",
    "dataset_target_folder": "root/image",
    "dataset_captions_json": "root/prompt.json",
    "wandb_key": "",
    "resolution": 512,
    "save_memory": False,
    "batch_size": 3,
    "image_logger_disabled": False,
    "image_logger_freq": 250,
    "log_every_n_steps": 1,
    "learning_rate": 1e-5,
    "max_steps": 9000,
    "max_epochs": 10,
    "save_ckpt_every_n_steps": 250,
    "save_top_k": -1,
    "save_weights_only": False,
    "save_last": False,
    "sd_locked": True,
    "only_mid_control": False,
    "gradient_accumulation_steps": 1,
    "wipe_older_ckpts": True,
    "multi_gpu": False,
}