File size: 1,237 Bytes
c54949e 14fcd3f c54949e 7cbde35 c54949e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
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"
|