---
job: extension
config:
  name: "spock_lora_v1"
  process:
    - type: 'sd_trainer'
      training_folder: "D:/AI_Training/spock_lora/output"
      device: cuda:0
      trigger_word: "spckmn"
      network:
        type: "lora"
        linear: 16
        linear_alpha: 16
      save:
        dtype: float16
        save_every: 250
        max_step_saves_to_keep: 4
        push_to_hub: false
      datasets:
        - folder_path: "D:/AI_Training/spock_lora/dataset"
          caption_ext: "txt"
          caption_dropout_rate: 0.05
          shuffle_tokens: false
          cache_latents_to_disk: true
          resolution: [512, 768]
      train:
        batch_size: 1
        steps: 1500
        gradient_accumulation_steps: 1
        train_unet: true
        train_text_encoder: false
        gradient_checkpointing: true
        noise_scheduler: "flowmatch"
        optimizer: "adamw8bit"
        lr: 1e-4
        dtype: bf16
        ema_config:
          use_ema: false
          ema_decay: 0.99
      model:
        name_or_path: "D:/AI_Training/hf_cache/hub/models--Niansuh--FLUX.1-schnell/snapshots/e69accbbbb51bd4722bc6ec9a3a2fa620e7ce7e5"
        assistant_lora_path: "ostris/FLUX.1-schnell-training-adapter"
        is_flux: true
        quantize: false
        low_vram: true
      sample:
        sampler: "flowmatch"
        sample_every: 250
        width: 512
        height: 512
        prompts:
          - "photo of spckmn, young man, neutral expression, studio lighting, white background"
          - "photo of spckmn, young man, wearing a black t-shirt, soft natural light"
          - "photo of spckmn, young man, pensive look, dark background, cinematic lighting"
        neg: ""
        seed: 42
        walk_seed: true
        guidance_scale: 1
        sample_steps: 4
meta:
  name: "[name]"
  version: '1.0'