MiniMax Lora Config

---
job: "extension"
config:
  name: "MiniMax VTX CGL 2"
  process:
    - type: "diffusion_trainer"
      training_folder: "/app/ai-toolkit/output"
      sqlite_db_path: "./aitk_db.db"
      device: "cuda"
      trigger_word: "penis"
      performance_log_every: 10
      network:
        type: "lora"
        linear: 16
        linear_alpha: 16
        conv: 16
        conv_alpha: 16
        lokr_full_rank: true
        lokr_factor: -1
        network_kwargs:
          ignore_if_contains:
            - "adaln_proj"
      save:
        dtype: "bf16"
        save_every: 250
        max_step_saves_to_keep: 4
        save_format: "diffusers"
        push_to_hub: false
      datasets:
        - folder_path: "/app/ai-toolkit/datasets/focused"
          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
          flip_x: false
          flip_y: false
          num_repeats: 1
          do_i2v: false
          fps: 24
          num_frames: 39
          auto_frame_count: true
      train:
        batch_size: 1
        bypass_guidance_embedding: false
        steps: 4000
        gradient_accumulation: 1
        train_unet: true
        train_text_encoder: false
        gradient_checkpointing: true
        noise_scheduler: "flowmatch"
        optimizer: "adamw8bit"
        timestep_type: "shift"
        content_or_style: "content"
        optimizer_params:
          weight_decay: 0.0001
        unload_text_encoder: false
        cache_text_embeddings: true
        lr: 0.0001
        ema_config:
          use_ema: false
          ema_decay: 0.99
        skip_first_sample: true
        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"
        do_guidance_loss: true
        guidance_loss_target: 3
        do_differential_guidance: true
        differential_guidance_scale: 3
        audio_loss_multiplier: 1
      logging:
        log_every: 1
        use_ui_logger: true
      model:
        name_or_path: "Comfy-Org/MiniMax-H3"
        quantize: true
        qtype: "convrot8"
        quantize_te: true
        qtype_te: "nvfp4"
        arch: "minimax_h3"
        low_vram: false
        model_kwargs: {}
        compile: false
        layer_offloading: false
        layer_offloading_text_encoder_percent: 1
        layer_offloading_transformer_percent: 1
        assistant_lora_path: "ostris/minimax_h3_training_adapter/minimax_h3_training_adapter_alpha.safetensors"
      sample:
        sampler: "flowmatch"
        sample_every: 250
        sample_start_step: 0
        width: 768
        height: 768
        samples:
          - prompt: "A nude man walking towards the camera, he has medium lenght brown hair and a beard. set in a lush forrest. His penis sways naturally with every step."
          - prompt: "A fully naked muscular man dancing  in the rain in the middle of a deserted street, he is blonde. His skin wet an glistening. As he moves his hips, his penis sways with his movementes."
          - prompt: "inside a brightly lit, modern shower. The walls are finished with large off-white tiles, accented by a decorative horizontal band of thin, multi-colored tiles in shades of grey, brown, and tan. A clear glass door with a simple metal handle encloses the space. The central figure is a young man with an exceptionally muscular and toned physique. He is shown from the neck down to his mid-thighs, standing in the center of the frame. He is fully nude, and his flaccid penis is visible. His arms are raised, and his hands are positioned near his head as water flows over his wet skin. He has several tattoos, including a large, detailed bird of prey on his right upper chest and shoulder, along with several smaller silhouettes of birds in flight trailing down his right side. The overall lighting is bright and direct, accentuating the definition of his muscles. [Action] The man takes a shower, his flaccid penis sways with his movements. He looks at the camera and says \"Yeah, almost done!\""
          - prompt: "set in an interior room, likely a bedroom or dressing area, illuminated by bright natural light coming through a window on the left. The window is covered with thin, translucent white curtains that diffuse the light and create a strong backlighting effect. To the right, a closet space is visible with various garments hanging on a rack above a small white dresser with a single dark knob. The central figure is a man with a lean and athletic build, shown in profile as he stands facing the window. He is fully nude, with the contours of his muscular arm, back, and buttocks highlighted by the light. Due to the high-contrast lighting, the front of his body is cast in shadow, and the overall scene has a soft, muted color palette. The man turns around, exposing his nude body and penis. He says \"Oh, hey. I didn't see you there.\""
          - prompt: "A muscular 28 year old man walking down the stairs in a luxurious appartment, he is fully naked, his penis is visible."
        neg: ""
        seed: 55
        walk_seed: true
        guidance_scale: 1
        sample_steps: 28
        num_frames: 107
        fps: 24
meta:
  name: "[name]"
  version: "1.0"