---
job: "extension"
config:
  name: "lora_name"
  process:
    - type: "sd_trainer"
      training_folder: "/app/ai-toolkit/output"
      device: "cuda"
      trigger_word: "trigger_word"
      network:
        type: "lora"
        linear: 16
        linear_alpha: 32
      save:
        dtype: "float16"
        save_every: 250
        max_step_saves_to_keep: 4
      datasets:
        - folder_path: "/app/ai-toolkit/datasets/data"
          caption_ext: "txt"
          caption_dropout_rate: 0.05
          num_frames: 41
          frame_rate: 24
          resolution:
            - 544
          is_video: true
      train:
        batch_size: 1
        steps: 2000
        gradient_accumulation: 1
        train_unet: true
        train_text_encoder: false
        gradient_checkpointing: true
        checkpointing_steps: 100
        noise_scheduler: "flowmatch"
        timestep_type: "linear"
        optimizer: "adamw8bit"
        lr: 0.0001
        optimizer_params:
          weight_decay: 0.0001
        dtype: "bf16"
        disable_sampling: true
        switch_boundary_every: 100
        cache_text_embeddings: true
        unload_text_encoder: true
      model:
        name_or_path: "ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16"
        arch: "wan22_14b"
        quantize: true
        qtype: "qfloat8"
        quantize_te: false
        qtype_te: "qfloat8"
        low_vram: false
        model_kwargs:
          train_high_noise: true
          train_low_noise: true
      sqlite_db_path: "./aitk_db.db"
      performance_log_every: 10
      logging:
        log_every: 1
        use_ui_logger: true