job: extension
config:
  name: "orbit_alpha_qwen21"
  process:
    - type: "sd_trainer"
      training_folder: "/output"
      device: cuda:0
      performance_log_every: 100
      network:
        type: "lora"
        linear: 32
        linear_alpha: 32
      save:
        dtype: "float16"
        save_every: 500
        max_step_saves_to_keep: 8
        push_to_hub: false
      logging:
        use_ui_logger: true
        log_every: 1
      datasets:
        - folder_path: "/train_data/targets"
          control_path: "/train_data/controls"
          caption_ext: "txt"
          resolution: [768]
          cache_latents_to_disk: true
      train:
        batch_size: 1
        steps: 2000
        gradient_accumulation: 1
        timestep_type: "weighted"
        train_unet: true
        train_text_encoder: false
        gradient_checkpointing: true
        noise_scheduler: "flowmatch"
        optimizer: "adamw8bit"
        lr: 1e-4
        dtype: bf16
        cache_text_embeddings: true
        skip_first_sample: true
        disable_sampling: true
      model:
        name_or_path: "Qwen/Qwen-Image-2.1"
        arch: "qwen_image_2"
        quantize: false
        quantize_te: false
        model_kwargs:
          rgba: true
          match_target_res: true
meta:
  name: "[name]"
  version: "1.0"
  description: "RGBA viewpoint-orbit LoRA for Qwen-Image-2.1; relative-azimuth orbit grammar; trained on Google Scanned Objects renders (CC-BY 4.0)"