job: custom_job
config:
  name: flux_train_replicate
  process:
  - type: custom_sd_trainer
    training_folder: output
    device: cuda:0
    trigger_word: zipsy
    network:
      type: lora
      linear: 32
      linear_alpha: 32
    save:
      dtype: float16
      save_every: 100
      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: 3000
      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: 100
      width: 1024
      height: 1024
      prompts:
      - This is a vibrant, digitally created anime-style illustration featuring a
        young woman named jean with a fair complexion and long, wavy blonde hair tied
        in a high ponytail with a blue ribbon. She has large, expressive blue eyes
        and a gentle, slightly blushing expression. Her outfit is a revealing, blue
        and white bikini with intricate, geometric patterns and a high neck. The bikini
        top is a halter style with a plunging neckline, accentuating her ample breasts.
        A blue gemstone pendant hangs from her neck, adding a touch of elegance. The
        background is a bright, sunny beach scene with a clear blue sky and gentle
        waves lapping at the shore. The water is a deep blue, and there are faintly
        visible green hills in the distance. The sand is light beige, and there are
        small, scattered white seashells near her feet. The woman's pose is slightly
        forward-leaning, with her right hand touching her hair and her left hand resting
        on her thigh. Her body language exudes confidence and playfulness. The overall
        style is clean and polished, with meticulous attention to detail in her hair,
        skin texture, and the vibrant colors of her outfit, which contrast beautifully
        with the serene beach setting.
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 3.5
      sample_steps: 28
meta:
  name: flux_train_replicate
  version: '1.0'