job: extension
config:
  name: arisu_tachibana_lora
  process:
  - type: diffusion_trainer
    training_folder: /app/ai-toolkit/output
    sqlite_db_path: /app/ai-toolkit/aitk_db.db
    device: cuda
    trigger_word: zzArisuTachibana
    performance_log_every: 10
    network:
      type: lora
      linear: 16
      linear_alpha: 17
      conv: 16
      conv_alpha: 16
      lokr_full_rank: true
      lokr_factor: -1
      network_kwargs:
        ignore_if_contains: []
    save:
      dtype: bf16
      save_every: 250
      max_step_saves_to_keep: 12
      save_format: diffusers
      push_to_hub: false
    datasets:
    - folder_path: /app/ai-toolkit/datasets/arisu_tachibana_dataset
      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
      controls: []
      shrink_video_to_frames: true
      num_frames: 1
      do_i2v: true
      flip_x: false
      flip_y: false
    train:
      batch_size: 2
      bypass_guidance_embedding: false
      steps: 3000
      gradient_accumulation: 1
      train_unet: true
      train_text_encoder: false
      gradient_checkpointing: true
      noise_scheduler: flowmatch
      optimizer: adamw8bit
      timestep_type: weighted
      content_or_style: style
      optimizer_params:
        weight_decay: 0.0001
      unload_text_encoder: false
      cache_text_embeddings: false
      lr: 0.0001
      ema_config:
        use_ema: false
        ema_decay: 0.99
      skip_first_sample: false
      force_first_sample: false
      disable_sampling: false
      dtype: bf16
      diff_output_preservation: false
      diff_output_preservation_multiplier: 1
      diff_output_preservation_class: person
      switch_boundary_every: 1
      loss_type: mse
    logging:
      log_every: 1
      use_ui_logger: true
    model:
      name_or_path: Tongyi-MAI/Z-Image-Turbo
      quantize: false
      qtype: qfloat8
      quantize_te: false
      qtype_te: qfloat8
      arch: zimage:turbo
      low_vram: false
      model_kwargs: {}
      layer_offloading: false
      layer_offloading_text_encoder_percent: 1
      layer_offloading_transformer_percent: 1
      assistant_lora_path: ostris/zimage_turbo_training_adapter/zimage_turbo_training_adapter_v2.safetensors
    sample:
      sampler: flowmatch
      sample_every: 250
      width: 1024
      height: 1024
      samples:
      - prompt: digital illustration, white background, anime, close-up shot, zzArisuTachibana,
          blue hair bow, sidelocks, blue dress, short sleeves, white collar, buttons
      - prompt: digital illustration, white background, anime, zzArisuTachibana, blue
          hair bow, sidelocks, blue dress, pleated skirt, short sleeves, white collar,
          buttons, brown belt
      - prompt: digital illustration, colorful cute background, anime, zzArisuTachibana,
          blue hair bow, sidelocks, blue dress, pleated skirt, short sleeves, white
          collar, buttons, brown belt, white socks, brown loafers,
        width: 832
        height: 1216
      - prompt: amazing anime-style illustration, upper-body shot, zzArisuTachibana,
          blue hair bow, sidelocks, blue dress, short sleeves, white collar, buttons,
          smile, sea background, sunset, cinematic lighting, masterpiece, best quality
      - prompt: digital art, anime style, zzArisuTachibana, blue hair bow, sidelocks,
          blue dress, pleated skirt, short sleeves, white collar, buttons, brown belt,
          blush, smile, wink, water, own hands together, water ripples, cinematic
          lighting, masterpiece, best quality
        width: 832
        height: 1216
      - prompt: masterpiece, best quality, amazing digital art, anime style, full
          body, zzArisuTachibana, blue hair bow, sidelock, blue dress, pleated skirt,
          short sleeves, white collar, buttons, brown belt, white socks, brown loafers,
          naughty, heavy snowy street, sunlight, cinematic lighting, volumetric lighting,
          atmospheric glow, complex contrast, detailed background, masterpiece, best
          quality
        width: 832
        height: 1216
      - prompt: amazing anime-style illustration, upper-body shot, zzArisuTachibana,
          blue hair bow, sidelocks, blue dress, short sleeves, white collar, buttons,
          pout, park, aesthetic cafe, morning sunlight, cinematic lighting, masterpiece,
          best quality
      - prompt: An anime-style amazing upper-body illustration of zzArisuTachibana
          against white background. She has blue hair bow and sidelock. She is wearing
          her white short-sleeved gym shirt with red trim, paired with red gym shorts.
      - prompt: zzArisuTachibana, She stands in a vertical anime-style illustration
          wearing a devil-themed costume against a pale background. Two small, curved
          horns protrude from a black headband, and a red bow secures her hair on
          the left side. Her mouth opens slightly. Her dress features a black and
          purple bodice with a vertically striped chest panel, off-the-shoulder purple
          sleeves with ruffled cuffs, and a corset-style waist laced in purple. A
          layered skirt with jagged, white-trimmed black points over purple fabric.
          Small bat wings extend from her back. She grips the shaft of a black trident
          behind her, its three prongs visible on the left. Dark thigh-high stockings
          held by garter straps lead down to red Mary Jane shoes accented with ankle
          wings. She stands on abstract patches of orange and yellow. Gold pixelated
          text in the bottom left reads "HAPPY HALLOWEEN 2099".
      - prompt: an asian girl, photorealistic, smile, looking at viewer
      - prompt: girl, playing chess at the park, bomb going off in the background
      - prompt: anime style illustration, a boy showing off his cool new t shirt at
          the beach, a shark is jumping out of the water in the background
      neg: ''
      seed: 42
      walk_seed: false
      guidance_scale: 1
      sample_steps: 8
      num_frames: 1
      fps: 1
meta:
  name: arisu_tachibana_v07-01
  version: '1.0'