config.yaml

job: custom_job
config:
  name: flux_train_replicate
  process:
  - type: custom_sd_trainer
    training_folder: output
    device: cuda:0
    trigger_word: GREMPLINSTYLE
    network:
      type: lora
      linear: 128
      linear_alpha: 128
    save:
      dtype: float16
      save_every: 250
      max_step_saves_to_keep: 1
    datasets:
    - folder_path: input_images
      caption_ext: txt
      caption_dropout_rate: 0.15
      shuffle_tokens: false
      cache_latents_to_disk: false
      cache_latents: true
      resolution:
      - 512
      - 768
      - 1024
    train:
      batch_size: 1
      steps: 6000
      gradient_accumulation_steps: 1
      train_unet: true
      train_text_encoder: false
      content_or_style: balanced
      gradient_checkpointing: false
      noise_scheduler: flowmatch
      optimizer: adamw8bit
      lr: 0.0001
      ema_config:
        use_ema: true
        ema_decay: 0.99
      dtype: bf16
    model:
      name_or_path: FLUX.1-dev
      is_flux: true
      quantize: false
    sample:
      sampler: flowmatch
      sample_every: 250
      width: 1024
      height: 1024
      prompts:
      - a GREMPLINSTYLE digital illustration of a character portrait of a blue dolphin
        man who is an accountant
      - an exhausted GREMPLINSTYLE digital illustration portrait of a bear man who
        is hung over and wearing a fleece sweater
      - a cute GREMPLINSTYLE digital illustration portrait of a cat woman who is yelling
        about stock options in a khaki suit
      - "Bust-length portrait of a female GREMPLINSTYLE teal-furred cat humanoid character\
        \ with triangular ears, mismatched eyes, and a short green bob, wearing a\
        \ pink puffy-sleeved blouse adorned with a small bow. Drawn in a retro MS\
        \ Paint\u2013style with bold black outlines, the character floats at chest\
        \ level against a purple halftone background, with no arms or lower body in\
        \ frame."
      - "GREMPLINSTYLE portrait digital illustration of a male purple bat humanoid\
        \ with large pink checkered ears, orange eyes, and small fangs, wearing a\
        \ pink bow tie and collar. A few small black flies hover around his head.\
        \ Illustrated in a retro MS Paint\u2013style with bold black outlines, the\
        \ figure appears slightly floating at chest with no visible arms or lower\
        \ body as an animated bust portrait."
      - A GREMPLINSTYLE ugly rat woman portrait
      - "GREMPLINSTYLE digital ms paint oekaki style edgy digital drawing of a male\
        \ green-skinned rat humanoid with a long pink-tipped nose, blond hair, and\
        \ oversized red-lensed sunglasses, wearing a simple black turtleneck. Illustrated\
        \ in a retro MS Paint\u2013style with bold black outlines, the figure appears\
        \ slightly floating at chest level against no background, with no visible\
        \ arms or lower body."
      - A pixelated GREMPLINSTYLE portrait illustration of a character with a gray
        face and wild blue spiky hair. The character has asymmetrical orange/yellow
        eyes. A prominent orange nose extends down the face like an exclamation mark
        with a dot at the bottom. The character has a sinister grin showing sharp
        teeth and is wearing a pale yellow shirt that buttons and a black apron with
        a bright pink bow tie as if working at an old time retro coffee shop. The
        overall aesthetic is deliberately crude and unsettling, rendered in a low-resolution
        pixel art style with bold outlines and limited palette.
      - Hand drawn oekaki ms paint pixelated GREMPLINSTYLE digital illustration floating
        bust portrait of an older female hacker cat on a checkered background
      - Hand drawn oekaki ms paint pixelated GREMPLINSTYLE digital illustration floating
        bust portrait of a cute cat
      - GREMPLINSTYLE Portrait of a garbageman goose. He wears a fucking sweater his
        grandma knit him
      - GREMPLINSTYLE Digital portrait resembling a slightly offbeat Oekaki drawing
        of a humanoid Housefly. The character is elderly, male, and dresses like an
        old-timey ringmaster/gambler. He wears a fancy pink bow tie combined with
        suspenders. He has sharp fangs and fly eyes as well as a big nose. Two flies
        flit around his head and the background is a dotted blue background.",
      - Oekaki chunky kiwibird lad in sloppy overalls screaming about crypto in the
        GREMPLINSTYLE style
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 3.5
      sample_steps: 28
meta:
  name: flux_train_replicate
  version: '1.0'