Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- job: extension
- config:
- name: Targeted_14_29_
- process:
- - type: diffusion_trainer
- training_folder: /home/ai-toolkit/output
- sqlite_db_path: /home/ai-toolkit/aitk_db.db
- device: cuda
- trigger_word: null
- performance_log_every: 10
- network:
- type: lora
- linear: 32
- linear_alpha: 32
- conv: 16
- conv_alpha: 16
- lokr_full_rank: true
- lokr_factor: -1
- network_kwargs:
- only_if_contains:
- - layers.14.attention.to_q
- - layers.14.attention.to_k
- - layers.14.attention.to_v
- - layers.14.attention.to_out.0
- - layers.14.feed_forward.w2
- - layers.14.feed_forward.w3
- - layers.15.attention.to_q
- - layers.15.attention.to_k
- - layers.15.attention.to_v
- - layers.15.attention.to_out.0
- - layers.15.feed_forward.w2
- - layers.15.feed_forward.w3
- - layers.16.attention.to_q
- - layers.16.attention.to_k
- - layers.16.attention.to_v
- - layers.16.attention.to_out.0
- - layers.16.feed_forward.w2
- - layers.16.feed_forward.w3
- - layers.17.attention.to_q
- - layers.17.attention.to_k
- - layers.17.attention.to_v
- - layers.17.attention.to_out.0
- - layers.17.feed_forward.w2
- - layers.17.feed_forward.w3
- - layers.18.attention.to_q
- - layers.18.attention.to_k
- - layers.18.attention.to_v
- - layers.18.attention.to_out.0
- - layers.18.feed_forward.w2
- - layers.18.feed_forward.w3
- - layers.19.attention.to_q
- - layers.19.attention.to_k
- - layers.19.attention.to_v
- - layers.19.attention.to_out.0
- - layers.19.feed_forward.w2
- - layers.19.feed_forward.w3
- - layers.20.attention.to_q
- - layers.20.attention.to_k
- - layers.20.attention.to_v
- - layers.20.attention.to_out.0
- - layers.20.feed_forward.w2
- - layers.20.feed_forward.w3
- - layers.21.attention.to_q
- - layers.21.attention.to_k
- - layers.21.attention.to_v
- - layers.21.attention.to_out.0
- - layers.21.feed_forward.w2
- - layers.21.feed_forward.w3
- - layers.22.attention.to_q
- - layers.22.attention.to_k
- - layers.22.attention.to_v
- - layers.22.attention.to_out.0
- - layers.22.feed_forward.w2
- - layers.22.feed_forward.w3
- - layers.23.attention.to_q
- - layers.23.attention.to_k
- - layers.23.attention.to_v
- - layers.23.attention.to_out.0
- - layers.23.feed_forward.w2
- - layers.23.feed_forward.w3
- - layers.24.attention.to_q
- - layers.24.attention.to_k
- - layers.24.attention.to_v
- - layers.24.attention.to_out.0
- - layers.24.feed_forward.w2
- - layers.24.feed_forward.w3
- - layers.25.attention.to_q
- - layers.25.attention.to_k
- - layers.25.attention.to_v
- - layers.25.attention.to_out.0
- - layers.25.feed_forward.w2
- - layers.25.feed_forward.w3
- - layers.26.attention.to_q
- - layers.26.attention.to_k
- - layers.26.attention.to_v
- - layers.26.attention.to_out.0
- - layers.26.feed_forward.w2
- - layers.26.feed_forward.w3
- - layers.27.attention.to_q
- - layers.27.attention.to_k
- - layers.27.attention.to_v
- - layers.27.attention.to_out.0
- - layers.27.feed_forward.w2
- - layers.27.feed_forward.w3
- - layers.28.attention.to_q
- - layers.28.attention.to_k
- - layers.28.attention.to_v
- - layers.28.attention.to_out.0
- - layers.28.feed_forward.w2
- - layers.28.feed_forward.w3
- - layers.29.attention.to_q
- - layers.29.attention.to_k
- - layers.29.attention.to_v
- - layers.29.attention.to_out.0
- - layers.29.feed_forward.w2
- - layers.29.feed_forward.w3
- save:
- dtype: bf16
- save_every: 250
- max_step_saves_to_keep: 20
- save_format: diffusers
- push_to_hub: false
- datasets:
- - folder_path: /home/ai-toolkit/datasets/example
- 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:
- - 768
- controls: []
- shrink_video_to_frames: true
- num_frames: 1
- do_i2v: true
- flip_x: false
- flip_y: false
- train:
- batch_size: 1
- bypass_guidance_embedding: false
- steps: 5000
- gradient_accumulation: 1
- train_unet: true
- train_text_encoder: false
- gradient_checkpointing: true
- noise_scheduler: flowmatch
- optimizer: adamw8bit
- timestep_type: sigmoid
- content_or_style: balanced
- optimizer_params:
- weight_decay: 0.0001
- unload_text_encoder: false
- cache_text_embeddings: true
- lr: 0.00025
- 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
- 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: a beaituful woman with heavy dark makeup is looking at the camera from a close angle, studio lighting.
- width: 768
- height: 1120
- neg: ''
- seed: 42
- walk_seed: true
- guidance_scale: 1
- sample_steps: 8
- num_frames: 1
- fps: 1
- logging:
- log_every: 1
- use_ui_logger: true
- meta:
- name: Targeted_14_29_
- version: '1.0'
Advertisement
Add Comment
Please, Sign In to add comment