job: custom_job
config:
  name: flux_train_replicate
  process:
  - type: custom_sd_trainer
    training_folder: output
    device: cuda:0
    trigger_word: KRHN
    network:
      type: lora
      linear: 16
      linear_alpha: 16
    save:
      dtype: float16
      save_every: 300
      max_step_saves_to_keep: 1
    datasets:
    - folder_path: input_images
      caption_ext: txt
      caption_dropout_rate: 0.05
      shuffle_tokens: false
      cache_latents_to_disk: false
      cache_latents: true
      resolution:
      - 512
      - 768
      - 1024
    train:
      batch_size: 1
      steps: 2000
      gradient_accumulation_steps: 1
      train_unet: true
      train_text_encoder: false
      content_or_style: balanced
      gradient_checkpointing: true
      noise_scheduler: flowmatch
      optimizer: adamw8bit
      lr: 0.0004
      ema_config:
        use_ema: true
        ema_decay: 0.99
      dtype: bf16
    model:
      name_or_path: FLUX.1-dev
      is_flux: true
      quantize: true
    sample:
      sampler: flowmatch
      sample_every: 300
      width: 1024
      height: 1024
      prompts:
      - a photo of KRHN, capture a mystical, high-contrast photograph of a stealthy
        ninja woman posing elegantly within the intricately carved, lantern-lit chambers
        of a centuries-old chinese temple. she stands poised, one hand grasping the
        hilt of her katana, its scabbard adorned with crimson tassels, while her piercing
        orange eyes seem to hold secrets of the ancient arts. the warm, golden light
        of lanterns dancing across the walls casts dramatic shadows, illuminating
        the ornate tapestries and forgotten relics scattered throughout the chamber,
        transporting the viewer to a realm of mystique and intrigue.
      - 'a photo of KRHN, capture a stunning, high-contrast photograph of a stealthy
        female ninja standing majestically on the lunar surface, illuminated by the
        radiant earthshine. her sleek black ninja garb blends seamlessly with the
        shadows, except for the subtle, crescent moon emblem on her chest, which glows
        softly like a beacon. her piercing orange eyes seem to pierce through the
        darkness, fixed intently on some distant target. in the background, the curvature
        of the earth rises like a glowing blue horizon, surrounded by the endless
        expanse of starry space, with the milky way streaking across the sky like
        a celestial banner. the ninja''s stance is poised and ready, as if prepared
        to leap into action at a moment''s notice, her slender figure silhouetted
        against the vibrant, dreamlike backdrop of our home planet. composition: low-angle
        shot, looking up at the ninja from the lunar surface, with the earth and stars
        stretching out above her.'
      - a photo of KRHN
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 3.5
      sample_steps: 28
meta:
  name: flux_train_replicate
  version: '1.0'