config.yaml

job: custom_job
config:
  name: flux_train_replicate
  process:
  - type: custom_sd_trainer
    training_folder: output
    device: cuda:0
    trigger_word: Zara
    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 modern Zara store exterior located in a bustling urban street, showcasing
        large glass windows and sleek minimalist architecture. The store''s illuminated
        signage glows prominently, reflecting off the polished surfaces. A blend of
        warm sunlight and artificial lights creates a sophisticated and inviting ambiance.
        Shoppers are seen entering and exiting, dressed in trendy outfits. The surrounding
        area features cobblestone walkways, streetlights, and a few potted plants,
        emphasizing the store''s upscale aesthetic: in the style of Zara Store.'
      - 'A spacious and stylish Zara store interior focused on the men''s clothing
        section. The space is defined by clean, neutral tones, including white walls,
        polished gray floors, and metallic clothing racks arranged neatly. Spotlights
        emphasize a variety of elegant blazers, tailored trousers, and casual wear.
        Mannequins are positioned strategically, dressed in contemporary fashion.
        A minimalist seating area with modern chairs and a small table is visible
        in the background: in the style of Zara Store.'
      - 'The interior of a vibrant Zara store showcasing the women''s clothing section.
        Rows of trendy garments hang from sleek racks, organized by color and style.
        Bright lighting highlights the luxurious feel of the space, with large mirrors
        and a polished marble floor adding a touch of sophistication. Shoppers are
        browsing the latest collections, while accessories like handbags and shoes
        are displayed on chic pedestals. A fitting room area with neutral curtains
        is subtly visible in the background: in the style of Zara Store.'
      - "A nighttime view of a Zara store entrance glowing with soft, warm lights\
        \ from within. The large glass fa\xE7ade reveals glimpses of neatly arranged\
        \ clothing racks and mannequins dressed in elegant seasonal outfits. The street\
        \ in front is lively with pedestrians and features clean, modern urban design\
        \ elements. The store\u2019s name is prominently displayed in sleek lettering,\
        \ creating an inviting and stylish atmosphere for late-night shoppers: in\
        \ the style of Zara Store."
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 3.5
      sample_steps: 28
meta:
  name: flux_train_replicate
  version: '1.0'