job: extension
config:
  name: new_ai_anime_v2
  process:
  - type: ui_trainer
    training_folder: /home/x/ai-toolkit-qwen/output
    sqlite_db_path: /home/x/ai-toolkit-qwen/aitk_db.db
    device: cuda
    trigger_word: null
    performance_log_every: 10
    network:
      type: lora
      linear: 8
      linear_alpha: 8
      conv: 16
      conv_alpha: 16
      lokr_full_rank: true
      lokr_factor: -1
      network_kwargs:
        ignore_if_contains: []
    save:
      dtype: bf16
      save_every: 50
      max_step_saves_to_keep: 2000
      save_format: safetensors
      push_to_hub: false
    datasets:
    - folder_path: /home/x/ai-toolkit-qwen/datasets/new_ai_anime
      control_path: null
      mask_path: null
      mask_min_value: 0.1
      default_caption: ''
      caption_ext: txt
      caption_dropout_rate: 0.05
      cache_latents_to_disk: true
      is_reg: false
      network_weight: 1
      resolution:
      - 512
      - 768
      - 1024
      controls: []
      shrink_video_to_frames: true
      num_frames: 1
      do_i2v: true
    train:
      batch_size: 1
      bypass_guidance_embedding: false
      steps: 10000
      gradient_accumulation: 1
      train_unet: true
      train_text_encoder: false
      gradient_checkpointing: true
      noise_scheduler: flowmatch
      optimizer: prodigy
      timestep_type: sigmoid
      content_or_style: balanced
      optimizer_params:
        weight_decay: 0
      unload_text_encoder: true
      cache_text_embeddings: true
      lr: 1
      ema_config:
        use_ema: false
        ema_decay: 0.99
      skip_first_sample: false
      disable_sampling: true
      dtype: bf16
      diff_output_preservation: false
      diff_output_preservation_multiplier: 1
      diff_output_preservation_class: person
    model:
      name_or_path: Qwen/Qwen-Image
      quantize: true
      qtype: uint3|ostris/accuracy_recovery_adapters/qwen_image_torchao_uint3.safetensors
      quantize_te: true
      qtype_te: qfloat8
      arch: qwen_image
      low_vram: true
      model_kwargs: {}
    sample:
      sampler: flowmatch
      sample_every: 50000
      width: 1024
      height: 1024
      samples:
      - prompt: Fine art; vibrant semi-realistic anime style digital painting; realistic
          shading. This digital illustration depicts a summer railroad track winding
          through the rolling hills of rural West Virginia. The steel rails curve
          gently through dense, lush Appalachian forest with thick stands of oak,
          maple, and pine trees. The tracks are bordered by wild summer vegetation
          - tall grasses, blackberry bushes, and colorful wildflowers like black-eyed
          Susans and Queen Anne's lace. In the background, the distinctive ridges
          and hollows of the Appalachian Mountains rise under a bright blue summer
          sky with scattered cumulus clouds. Small wooden farmhouses with wraparound
          porches dot the landscape, surrounded by fields of tall corn and tobacco
          plants. The railroad crossing features rustic wooden crossing barriers and
          warning signs showing weathering from years of mountain weather. The atmosphere
          captures the humid, green essence of a West Virginia summer - the air feels
          thick with moisture and the scent of pine. Sunlight filters through the
          forest canopy, creating dappled patterns on the tracks and ground. The scene
          evokes a sense of isolation and natural beauty, with the railroad serving
          as a lifeline through the remote mountain terrain. Photorealistic style
          with emphasis on the rich greens of the Appalachian forest, the texture
          of weathered railroad infrastructure, and the distinctive topography of
          West Virginia's rolling hills and valleys.
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 4
      sample_steps: 25
      num_frames: 1
      fps: 1
meta:
  name: new_ai_anime_v2
  version: '1.0'