| name: test-run | |
| model_path: "runwayml/stable-diffusion-v1-5" | |
| train_epochs: 10 | |
| batch_size: 4 | |
| resolution: 512 | |
| seed: 1138 | |
| arb: | |
| enabled: true | |
| debug: false | |
| base_res: [512, 512] | |
| max_size: [768, 512] | |
| divisible: 64 | |
| max_ar_error: 4 | |
| min_dim: 256 | |
| dim_limit: 1024 | |
| dataset: | |
| img_path: "./imagenet/imagenet_images" | |
| egdemap_path: "./imagenet/edge_maps" | |
| captions_path: "./captions.json" | |
| center_crop: true | |
| ucg: 0.0 | |
| debug_arb: false | |
| num_workers: 3 | |
| process_tags: false | |
| cache_latents: false | |
| important_tags: [] | |
| allow_duplicates: false | |
| optimizer: | |
| name: bitsandbytes.optim.AdamW8bit | |
| params: | |
| lr: 2e-4 | |
| weight_decay: 1e-2 | |
| eps: 1e-8 | |
| monitor: | |
| wandb_id: "" | |
| huggingface_repo: "" | |
| huggingface_token: "" | |
| sampling: | |
| enabled: false | |
| use_wandb: true | |
| every_n_steps: -1 | |
| every_n_epochs: 1 | |
| save_dir: "samples" | |
| seed: 1139 | |
| height: 512 | |
| width: 512 | |
| steps: 20 | |
| cfg_scale: 9 | |
| negative_prompts: | |
| - "lowres, bad anatomy, text, error, extra digit, cropped" | |
| - "lowres, low quality" | |
| prompts: | |
| - "a girl running fast on the country lane" | |
| - "a girl in black serafuku standing in a field. Tags: solo, food, fruit, lemon, masterpiece" | |