Flux.1 Schnell Pvc Style Lora

job: extension
config:
  name: flux_lora_pvc_schnell_1
  process:
  - type: sd_trainer
    training_folder: output
    device: cuda:3
    network:
      type: lora
      linear: 16
      linear_alpha: 16
    save:
      dtype: bfloat16
      save_every: 250
      max_step_saves_to_keep: 10
    datasets:
    - folder_path: /workspace/ai-toolkit/dataset/pvc
      caption_ext: txt
      caption_dropout_rate: 0.01
      shuffle_tokens: true
      cache_latents_to_disk: true
      resolution:
      - 768
      - 1024
    train:
      batch_size: 4
      steps: 10000
      gradient_accumulation_steps: 1
      train_unet: true
      train_text_encoder: false
      gradient_checkpointing: true
      noise_scheduler: flowmatch
      optimizer: adamw8bit
      lr: 0.0001
      ema_config:
        use_ema: true
        ema_decay: 0.99
      dtype: bf16
    model:
      name_or_path: black-forest-labs/FLUX.1-schnell
      assistant_lora_path: ostris/FLUX.1-schnell-training-adapter
      is_flux: true
      quantize: true
    sample:
      sampler: flowmatch
      sample_every: 250
      width: 832
      height: 1152
      prompts:
      - 1girl, solo, blue hair, cat ears, parted bangs, long hair, looking at viewer,
        white dress shirt, rainy, wariza, sitting
      - 1girl, solo, full body, gloves, groin, hair between eyes, hair ornament, head
        tilt, holding, holding staff, horns, long hair, looking at viewer, maid headdress,
        mole, mole under eye, navel, parted lips, purple eyes, purple hair, purple
        theme, shawl, shorts, single horn, staff, standing, thighhighs, twintails,
        white footwear, white gloves, white thighhighs
      - 1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green
        background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair,
        solo, upper body, yellow shirt,
      - 1girl, bangs, bare shoulders, beret, black hair, black shorts, blue hair,
        bracelet, breasts, buttons, colored inner hair, double-breasted, eyewear removed,
        green headwear, green jacket, grey eyes, grey sky, hat, jacket, jewelry, long
        hair, looking at viewer, multicolored hair, neck ring, o-ring, off shoulder,
        rain, round eyewear, shorts, sidelocks, small breasts, solo, sunglasses, wavy
        hair, wet, zipper,
      - 1girl, brown hair, green eyes, colorful, autumn, cumulonimbus clouds, lighting,
        blue sky, falling leaves, garden
      - 1girl, bangs, black wings, blush, choker, collarbone, feathered wings, hair
        between eyes, halo, hand on hip, hat, head wings, long hair, long sleeves,
        looking at viewer, pink eyes, pink hair, pleated skirt, school hat, school
        uniform, serafuku, sidelocks, simple background, skirt, solo, twintails, wavy
        mouth, white background, wings, waifu,
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 1
      sample_steps: 4
    logging:
      use_wandb: true
      project_name: flux_lora_pvc_schnell_1
      run_name: run-1
meta:
  name: flux_lora_schnell_pvc
  version: '1.0'