config.yaml

job: extension
config:
  name: devsav
  process:
  - type: sd_trainer
    training_folder: output
    device: cuda:0
    trigger_word: devsav
    network:
      type: lora
      linear: 16
      linear_alpha: 16
    save:
      dtype: float16
      save_every: 500
      max_step_saves_to_keep: 10
    datasets:
    - folder_path: /root/train/
      caption_ext: txt
      caption_dropout_rate: 0.05
      shuffle_tokens: false
      cache_latents_to_disk: true
      resolution:
      - 512
      - 768
      - 1024
    train:
      batch_size: 1
      steps: 4001
      gradient_accumulation_steps: 1
      train_unet: true
      train_text_encoder: false
      gradient_checkpointing: true
      noise_scheduler: flowmatch
      optimizer: adamw8bit
      lr: 0.0001
      ema_config:
        use_ema: true
        ema_decay: 0.99
      dtype: bf16
    model:
      name_or_path: black-forest-labs/FLUX.1-dev
      is_flux: true
      quantize: true
    sample:
      sampler: flowmatch
      sample_every: 1
      width: 1024
      height: 1024
      prompts:
      - devsav 21-year-old man holding a coffee cup, in a beanie, sitting at a cozy
        cafe, winter morning
      - devsav 21-year-old man as a DJ at a night club, fish eye lens, smoke machine,
        laser lights, holding a martini, vibrant atmosphere
      - devsav 21-year-old man showing off his cool new t-shirt at the beach, a shark
        is jumping out of the water in the background, sunny day
      - devsav 21-year-old man building a log cabin in the snow-covered mountains,
        winter wonderland
      - devsav 21-year-old man playing the guitar, on stage, singing a song, laser
        lights, punk rocker, energetic performance
      - devsav 21-year-old hipster man with a beard, building a chair, in a wood shop,
        rustic environment
      - devsav 21-year-old man, white background, medium shot, modeling clothing,
        studio lighting, white backdrop, professional look
      - devsav 21-year-old man holding a sign that says, 'this is a sign', humorous
        expression
      - devsav 21-year-old man in a post-apocalyptic world, with a shotgun, in a leather
        jacket, in a desert, with a motorcycle, gritty scene
      - devsav 21-year-old man in a professional suit, giving a presentation, in a
        modern office, confident demeanor
      - devsav 21-year-old man in Indian attire, at a traditional wedding, holding
        a bouquet, colorful ceremony
      - devsav 21-year-old man with a cat, sitting on a couch, reading a book, cozy
        living room
      - devsav 21-year-old man feeding pigeons, at Trafalgar Square, London, iconic
        landmark
      - devsav 21-year-old man in a LinkedIn profile photo, smiling, in a business
        suit, white background, professional headshot
      - devsav 21-year-old man in a suit, shaking hands, in a boardroom, with a cityscape
        view, business deal
      - devsav 21-year-old man in Indian traditional dress, at a festival, dancing,
        joyful celebration
      - devsav 21-year-old man with a dog, hiking in the mountains, sunset, adventurous
        spirit
      - devsav 21-year-old man with a parrot, in a tropical rainforest, vibrant greenery
      - devsav 21-year-old man in a professional suit, working on a laptop, at a cafe,
        focused work
      - devsav 21-year-old man in Indian attire, cooking in a traditional kitchen,
        aromatic spices
      - devsav 21-year-old man with a horse, at a ranch, sunset, rustic lifestyle
      - devsav 21-year-old man in a kimono, walking in a Japanese garden, serene surroundings
      - devsav 21-year-old man in a professional suit, talking on the phone, in an
        office, busy day
      - devsav 21-year-old man in Indian attire, at a temple, praying, spiritual moment
      - devsav 21-year-old man with a tiger, in a jungle, wild encounter
      - devsav 21-year-old man with a dog, at the Eiffel Tower, Paris, romantic backdrop
      - devsav 21-year-old man in a professional suit, sitting at a desk, working
        on a computer, productive environment
      - devsav 21-year-old man in Indian attire, at a market, buying spices, vibrant
        marketplace
      - devsav 21-year-old man with a falcon, in a desert, sunset, majestic bird
      - devsav 21-year-old man with a panda, in a bamboo forest, peaceful setting
      - devsav 21-year-old man in a professional suit, giving an interview, on TV,
        articulate speaker
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 4
      sample_steps: 40
meta:
  name: devsav
  version: '1.0'