name: fulldan_cache

trainer:
  model_path: ../models/sdxl-fd5m-epoch-09.ckpt
  batch_size: 8
  resolution: 1024
  seed: 1138
  wandb_id: "yada-debug"
  use_ema: false
  use_xformers: true
  accumulate_grad_batches: 1
  gradient_clip_val: 0.0

  offset_noise: true
  offset_noise_val: 0.08
  min_snr: false
  min_snr_val: 5

  checkpoint_dir: checkpoint
  checkpoint_freq: 8
  checkpoint_steps: -1
  save_weights_only: true
  max_epochs: -1
  max_steps: -1
  
lightning:
  accelerator: gpu
  devices: -1
  precision: 16-mixed

cache:
  enabled: true
  cache_bsz: 4
  cache_dir: cache
  # Precompute embeddings for faster training (on every epoch)
  precompute_embeds: false
  use_legacy_key: false
  target: ["images"] #[images, prompts]

dataset:
  img_path: 
      - "../datasets/debug_set_2"
  tag_processor: data.processors.shuffle_prompts
  center_crop: false
  ucg: 0.1
  max_token_length: 225 # [75, 150, 225]
  debug_arb: false
  allow_duplicates: false

optimizer:
  name: prodigyopt.Prodigy
  params:
    lr: 1

scheduler:
  name: transformers.get_cosine_schedule_with_warmup
  params:
    num_warmup_steps: 100 
    num_training_steps: 5000
    last_epoch: -1

sampling:
  enabled: true
  use_wandb: true
  every_n_steps: -1
  every_n_epochs: 8
  save_dir: "samples"
  seed: 1024
  height: 1024
  width: 1024
  steps: 28
  cfg_scale: 8
  negative_prompts: 
    - "text, error, signature, watermark, username, realistic,3d, lowres, cropped, worth quality ,low quality, normal quality, jpeg artifacts, blurry, bad anatomy, bad hands, bad arms, bad feet"
    - "text, error, signature, watermark, username, realistic,3d, lowres, cropped, worth quality ,low quality, normal quality, jpeg artifacts, blurry, bad anatomy, bad hands, bad arms, bad feet"
    - "text, error, signature, watermark, username, realistic,3d, lowres, cropped, worth quality ,low quality, normal quality, jpeg artifacts, blurry, bad anatomy, bad hands, bad arms, bad feet"
    - "text, error, signature, watermark, username, realistic,3d, lowres, cropped, worth quality ,low quality, normal quality, jpeg artifacts, blurry, bad anatomy, bad hands, bad arms, bad feet"
    - "text, error, signature, watermark, username, realistic,3d, lowres, cropped, worth quality ,low quality, normal quality, jpeg artifacts, blurry, bad anatomy, bad hands, bad arms, bad feet"
    - "lowres, low quality, text, error, extra digit, cropped"
    - "text, error, signature, watermark, username, realistic,3d, lowres, cropped, worth quality ,low quality, normal quality, jpeg artifacts, blurry, bad anatomy, bad hands, bad arms, bad feet"
    - "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
    - "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
    - "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
    - "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
    - "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
    - "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
    - "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
  prompts: 
    - "best quality, a girl with a yellow hat and a yellow shirt, sitting, white legwear, uniform, jacket, thighhighs, bare shoulders, high contrast, paint splatter"
    - "best quality, 1girl, solo, loli, cat girl, silver hair ,blue eyes, flat chest, solo, beautiful detailed background, messy hair, long hair"
    - "masterpiece, best quality, 1girl, a clean painting of a girl with long white hair drinking coffee in a coffee bar"
    - "best quality, 1girl, a clean painting of a girl with long white hair drinking coffee in a coffee bar"
    - "1girl, a clean painting of a girl with long white hair drinking coffee in a coffee bar"
    - "1girl,sitting,fantasy,masterpiece,best quality,(long blonde hair),(blue eyes),(floating hair),(black ribbed sweater:1.1),(red plaid skirt:1.1),(cat ears)"
    - "A girl with collar, chain, and leash sits by a window, looking at the viewer. She has blue eyes, blonde hair, and small breasts, with cum visible on her body and hair. Her facial expression is one of tears and clenched teeth, and she wears metal cuffs and a metal collar."
    - "A girl with collar, chain, and leash sits by a window, looking at the viewer. She has blue eyes, blonde hair, and small breasts, with cum visible on her body and hair. Her facial expression is one of tears and clenched teeth, and she wears metal cuffs and a metal collar."
    - "masterpiece, best quality, 1girl, solo,loli,wedding dress, see-through highleg leotard, veil, elbow gloves, white thighhighs, crown, earrings, bow on waist, sideboob, lace,"
    - "1girl, solo, shirt, thighhighs, skirt, hands on hips, white shirt, crystal, flandre scarlet, blonde hair, grey skirt, red bow, red eyes, black thighhighs, wings, white background,"
    - "1girl, solo, hina_(blue_archive), long_hair, alternate_costume, horns, maid, purple_eyes, white_hair, demon_horns, wings"
    - "1girl, solo, stomach, halo, breasts, pink_hair, bikini, navel, swimsuit, white_bikini, jacket, long_hair"
    - "1girl, solo, forehead, long_hair, halo, white_background, red_hair, purple_eyes, maid, white_bow, maid_headdress"
    - "1girl, breasts, animal_ears, rabbit_ears, cup, solo, fake_animal_ears, looking_at_viewer, pantyhose, playboy_bunny, leotard, cleavage, teacup, brown_hair, bare_shoulders"

# arb config will be calculated automatically if not specified
# to override, uncomment and specify values
#
# arb:
#   enabled: false
#   debug: false
#   base_res: [512, 512]
#   max_size: [768, 512]
#   divisible: 64
#   max_ar_error: 1
#   min_dim: 256
#   dim_limit: 1024
#
# 
# lora:
#   enabled: true
#   rank: 16
#   lora_alpha: 8
#   multipier: 1.0
#   unet_lr: 1e-4