Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- ---
- job: "extension"
- config:
- name: "wan2.2_i2v_pbrstyle"
- process:
- - type: "diffusion_trainer"
- training_folder: "/mnt/t/ComfyUI_training/ai-toolkit-output"
- sqlite_db_path: "./aitk_db.db"
- device: "cuda"
- trigger_word: "PBRSty1e"
- performance_log_every: 10
- network:
- type: "lora"
- linear: 16
- linear_alpha: 16
- conv: 8
- conv_alpha: 8
- lokr_full_rank: true
- lokr_factor: -1
- network_kwargs:
- ignore_if_contains: []
- save:
- dtype: "bf16"
- save_every: 500
- max_step_saves_to_keep: 16
- save_format: "diffusers"
- push_to_hub: false
- datasets:
- - folder_path: "/mnt/c/ComfyUI_training/ai-toolkit-datasets/PBRStyleImg"
- mask_path: null
- mask_min_value: 0.1
- default_caption: ""
- caption_ext: "txt"
- caption_dropout_rate: 0.05
- cache_latents_to_disk: false
- is_reg: false
- network_weight: 1
- resolution:
- - 256
- controls: []
- shrink_video_to_frames: true
- num_frames: 1
- do_i2v: false
- flip_x: true
- flip_y: false
- - folder_path: "/mnt/c/ComfyUI_training/ai-toolkit-datasets/PBRStyleVid"
- mask_path: null
- mask_min_value: 0.1
- default_caption: ""
- caption_ext: "txt"
- caption_dropout_rate: 0.05
- cache_latents_to_disk: false
- is_reg: false
- network_weight: 0.5
- resolution:
- - 256
- controls: []
- shrink_video_to_frames: true
- num_frames: 18
- do_i2v: true
- flip_x: true
- flip_y: false
- 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: "adamw"
- timestep_type: "sigmoid"
- content_or_style: "style"
- optimizer_params:
- weight_decay: 0.0001
- unload_text_encoder: false
- cache_text_embeddings: false
- lr: 0.000015
- ema_config:
- use_ema: false
- ema_decay: 0.99
- skip_first_sample: true
- force_first_sample: false
- disable_sampling: true
- dtype: "fp16"
- diff_output_preservation: false
- diff_output_preservation_multiplier: 1
- diff_output_preservation_class: "person"
- switch_boundary_every: 10
- loss_type: "mse"
- model:
- name_or_path: "ai-toolkit/Wan2.2-I2V-A14B-Diffusers-bf16"
- quantize: true
- quantize_te: true
- arch: "wan22_14b_i2v"
- low_vram: true
- model_kwargs:
- train_high_noise: true
- train_low_noise: true
- layer_offloading: false
- layer_offloading_text_encoder_percent: 1
- layer_offloading_transformer_percent: 1
- qtype: "uint6"
- qtype_te: "qfloat8"
- sample:
- sampler: "flowmatch"
- sample_every: 1000
- width: 256
- height: 256
- samples:
- - prompt: ""
- neg: ""
- seed: 42
- walk_seed: true
- guidance_scale: 4
- sample_steps: 20
- num_frames: 41
- fps: 16
- logging:
- log_every: 1
- use_ui_logger: true
- meta:
- name: "[name]"
- version: "1.0"
Advertisement
Add Comment
Please, Sign In to add comment
