job: extension
config:
  name: human_training
  process:
  - type: sd_trainer
    training_folder: output
    device: cuda:0
    network:
      type: lora
      linear: 32
      linear_alpha: 16
    save:
      dtype: float16
      save_every: 1000
      max_step_saves_to_keep: 5
      push_to_hub: true
    hf_repo_id: linhqyy/human-hig-v4
    hf_private: false
    datasets:
    - folder_path: /workspace/human_training
      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: 5000
      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: false
    sample:
      sampler: flowmatch
      sample_every: 500
      width: 1024
      height: 1024
      prompts:
      - photo-realistic. wide angle view, An elderly monk with bald head and a long
        white beard playing a flute in a tranquil waterfall setting. relaxing face,
        plays a traditional wooden flute near his mouth, both hand placed on the flute
        and separated, bright the man face with natural light. He is wearing traditional
        yellow Taoist robes and is meditating on the mountain path. The background
        is a waterfall over a jungle, wide view. High Detail, Beautiful scenary, realistic
        light, soft light, realistic photography, taken by Sony a7300.
      - woman, female, outdoor, nature, forest, waterfall, serene, peaceful, musical
        instrument, handpan, long hair, wavy, floral dress, goddess-like, ethereal,
        sitting, golden hour, sunlight, water reflection, green tones, romantic, portrait,
        delicate, tranquil, magic realism, fantasy-like, flowy dress, wreath, ivy,
        idyllic, artistic, soft lighting, warm hues, bohemian, dreamy atmosphere,
        natural makeup, ambient light, harmonious, calm.
      - A young woman with soft, delicate features is shown sleeping calmly on a white
        pillow, wrapped in cozy blankets, wide view. The gentle blue hue of moonlight
        illuminates her serene face through the window beside her, casting a soft
        glow over the room. Her hair flows loosely on the pillow, adding to the tranquil
        mood. Outside the window, the deep blue night sky sets a calming, dreamlike
        atmosphere.
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 4
      sample_steps: 20
meta:
  name: human_training
  version: '1.0'