job: extension
config:
  name: human
  process:
  - type: sd_trainer
    training_folder: output
    device: cuda:0
    network:
      type: lora
      linear: 16
      linear_alpha: 16
    save:
      dtype: float16
      save_every: 1000
      max_step_saves_to_keep: 4
    datasets:
    - folder_path: /workspace/imgs
      caption_ext: txt
      caption_dropout_rate: 0.05
      shuffle_tokens: false
      cache_latents_to_disk: true
      resolution:
      - 1024
    train:
      batch_size: 1
      steps: 10000
      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: 1344
      height: 768
      prompts:
      - An illustration of, a young woman, lying down, with long, flowing brown hair,
        wearing a white dress, eyes closed, lips slightly parted, serene and peaceful
        expression, hand resting behind head, soft ambient lighting, night sky, background
        with blurred, dark structures, hints of nature and foliage, bright moonlight
        casting a gentle glow, realistic yet artistic style, high-resolution, extremely
        detailed.
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 4
      sample_steps: 20
meta:
  name: human
  version: '1.0'