job: custom_job
config:
  name: flux_train_replicate
  process:
  - type: custom_sd_trainer
    training_folder: output
    device: cuda:0
    trigger_word: westfield
    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 shopping center with modern architecture, featuring wide glass
        ceilings that allow natural light to flood in, surrounded by vibrant storefronts
        and shoppers carrying bags, at the Westfield Mall in Velizy.
      - An interior view of a trendy food court with stylish furniture, illuminated
        by soft ambient lighting, with groups of people enjoying meals and conversations,
        at the Westfield Mall in Velizy.
      - A sleek and contemporary storefront with mannequins in elegant outfits displayed
        in a brightly lit window, framed by polished marble flooring and a lively
        crowd passing by, at the Westfield Mall in Velizy.
      - A wide-angle shot of the central atrium with a grand staircase, featuring
        an impressive decorative centerpiece and holiday lights, as shoppers explore
        the mall in a cheerful atmosphere, at the Westfield Mall in Velizy.
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 3.5
      sample_steps: 28
meta:
  name: flux_train_replicate
  version: '1.0'