job: custom_job
config:
  name: flux_train_replicate
  process:
  - type: custom_sd_trainer
    training_folder: output
    device: cuda:0
    trigger_word: westfieldvelizy
    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 bustling indoor scene at Westfield Mall in Velizy, featuring vibrant shops
        with modern storefronts, fashionable shoppers carrying branded bags, and sleek
        architectural details like glass railings and polished floors. Bright lighting
        from skylights and ceiling fixtures creates a lively and energetic atmosphere,
        capturing the essence of a busy shopping day: in the style of westfieldvelizy.'
      - 'An elegant cafe inside Westfield Mall in Velizy, with modern seating arrangements,
        cozy lighting, and patrons enjoying artisanal coffee and desserts. The background
        shows stylish decor, a touch of greenery, and the buzz of mall activity, blending
        sophistication with a casual vibe: in the style of westfieldvelizy.'
      - 'A wide-angle shot of the central atrium in Westfield Mall, Velizy, showcasing
        its expansive space with large decorative elements, a water fountain, and
        escalators leading to various levels. People of all ages are walking, shopping,
        and relaxing, immersed in the vibrant and luxurious mall experience: in the
        style of westfieldvelizy.'
      - 'A winter-themed promotional event at Westfield Mall in Velizy, featuring
        a beautifully decorated area with fairy lights, a Christmas tree, and festive
        displays. Shoppers in coats and scarves enjoy the cheerful atmosphere, and
        a Santa Claus figure is interacting with children near a pop-up photo booth:
        in the style of westfieldvelizy.'
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 3.5
      sample_steps: 28
meta:
  name: flux_train_replicate
  version: '1.0'