job: custom_job
config:
  name: flux_train_replicate
  process:
  - type: custom_sd_trainer
    training_folder: output
    device: cuda:0
    trigger_word: zipsy_nai
    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.2
      shuffle_tokens: false
      cache_latents_to_disk: false
      cache_latents: true
      resolution:
      - 512
      - 768
      - 1024
    train:
      batch_size: 1
      steps: 1000
      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 digital anime-style illustration featuring a young woman with a
        fair complexion and long, flowing blonde hair. She has large, expressive blue
        eyes and a gentle smile. She is dressed in a revealing, form-fitting blue
        and white swimsuit that accentuates her ample bust and curves. The swimsuit
        has a high neckline with a keyhole design, and a blue ribbon around her neck.
        Her outfit is adorned with intricate white lace and blue accents, including
        a diamond-shaped cutout on her right thigh.She is standing on a sandy beach
        with a clear blue ocean behind her, where the water meets the sky. The sky
        is a vibrant blue with scattered white clouds, and the horizon features a
        faint outline of distant mountains. The woman's pose is relaxed, with her
        left hand touching her hair and her right hand resting on her hip. Her expression
        is confident and playful, adding to the overall cheerful and summery atmosphere
        of the scene.The illustration is highly detailed, with a focus on soft textures
        and vibrant colors, typical of high-quality digital anime art. The overall
        mood is light and carefree, capturing a moment of relaxation by the beach.
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 3.5
      sample_steps: 28
meta:
  name: flux_train_replicate
  version: '1.0'