job: custom_job
config:
  name: flux_train_replicate
  process:
  - type: custom_sd_trainer
    training_folder: output
    device: cuda:0
    trigger_word: Aigle
    network:
      type: lora
      linear: 16
      linear_alpha: 16
    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: 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:
      - "A charming Aigle store nestled in the French countryside, featuring a rustic\
        \ wooden fa\xE7ade with large glass windows displaying outdoor apparel and\
        \ boots. The signage is elegantly mounted above the entrance with the classic\
        \ Aigle logo in white against a forest-green background. Flower pots with\
        \ vibrant blooms line the entrance, and the store is surrounded by lush greenery\
        \ under a clear blue sky. A cozy bench sits by the door, inviting passersby."
      - "A sleek, modern Aigle store located on a bustling city street. The storefront\
        \ has floor-to-ceiling glass panels, showcasing premium outdoor gear and stylish\
        \ mannequins in hiking outfits. The store's logo is prominently illuminated\
        \ in white on a black background above the entrance. Shoppers and pedestrians\
        \ in urban attire walk past, reflecting the store\u2019s contemporary appeal.\
        \ The scene is set during golden hour, with sunlight casting a warm glow on\
        \ the building."
      - An inviting Aigle store situated in a quaint mountain town, surrounded by
        snow-capped peaks and pine trees. The storefront is built with natural stone
        and timber, featuring large windows displaying winter gear like boots, jackets,
        and scarves. The Aigle sign is carved into a wooden board above the entrance.
        Snow blankets the ground, and ski equipment is propped against the wall beside
        the door. Soft, warm light glows from inside, creating a cozy atmosphere.
      - A relaxed Aigle store located along a sandy beach promenade. The storefront
        is painted in soft pastel tones, with a weathered wood texture that reflects
        the seaside environment. Large windows showcase nautical-themed outdoor apparel,
        with ropes and fishing nets used as decorative props. The Aigle logo is etched
        onto a metal sign above the door. The ocean is visible in the background,
        with gentle waves lapping against the shore and palm trees swaying in the
        breeze. The scene is bathed in warm sunlight, evoking a vacation vibe.
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 3.5
      sample_steps: 28
meta:
  name: flux_train_replicate
  version: '1.0'