capitan01r

ZiT_config_targeted_14_29_rank_32

Apr 2nd, 2026 (edited)
57
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 6.32 KB | None | 0 0
  1. job: extension
  2. config:
  3. name: Targeted_14_29_
  4. process:
  5. - type: diffusion_trainer
  6. training_folder: /home/ai-toolkit/output
  7. sqlite_db_path: /home/ai-toolkit/aitk_db.db
  8. device: cuda
  9. trigger_word: null
  10. performance_log_every: 10
  11. network:
  12. type: lora
  13. linear: 32
  14. linear_alpha: 32
  15. conv: 16
  16. conv_alpha: 16
  17. lokr_full_rank: true
  18. lokr_factor: -1
  19. network_kwargs:
  20. only_if_contains:
  21. - layers.14.attention.to_q
  22. - layers.14.attention.to_k
  23. - layers.14.attention.to_v
  24. - layers.14.attention.to_out.0
  25. - layers.14.feed_forward.w2
  26. - layers.14.feed_forward.w3
  27. - layers.15.attention.to_q
  28. - layers.15.attention.to_k
  29. - layers.15.attention.to_v
  30. - layers.15.attention.to_out.0
  31. - layers.15.feed_forward.w2
  32. - layers.15.feed_forward.w3
  33. - layers.16.attention.to_q
  34. - layers.16.attention.to_k
  35. - layers.16.attention.to_v
  36. - layers.16.attention.to_out.0
  37. - layers.16.feed_forward.w2
  38. - layers.16.feed_forward.w3
  39. - layers.17.attention.to_q
  40. - layers.17.attention.to_k
  41. - layers.17.attention.to_v
  42. - layers.17.attention.to_out.0
  43. - layers.17.feed_forward.w2
  44. - layers.17.feed_forward.w3
  45. - layers.18.attention.to_q
  46. - layers.18.attention.to_k
  47. - layers.18.attention.to_v
  48. - layers.18.attention.to_out.0
  49. - layers.18.feed_forward.w2
  50. - layers.18.feed_forward.w3
  51. - layers.19.attention.to_q
  52. - layers.19.attention.to_k
  53. - layers.19.attention.to_v
  54. - layers.19.attention.to_out.0
  55. - layers.19.feed_forward.w2
  56. - layers.19.feed_forward.w3
  57. - layers.20.attention.to_q
  58. - layers.20.attention.to_k
  59. - layers.20.attention.to_v
  60. - layers.20.attention.to_out.0
  61. - layers.20.feed_forward.w2
  62. - layers.20.feed_forward.w3
  63. - layers.21.attention.to_q
  64. - layers.21.attention.to_k
  65. - layers.21.attention.to_v
  66. - layers.21.attention.to_out.0
  67. - layers.21.feed_forward.w2
  68. - layers.21.feed_forward.w3
  69. - layers.22.attention.to_q
  70. - layers.22.attention.to_k
  71. - layers.22.attention.to_v
  72. - layers.22.attention.to_out.0
  73. - layers.22.feed_forward.w2
  74. - layers.22.feed_forward.w3
  75. - layers.23.attention.to_q
  76. - layers.23.attention.to_k
  77. - layers.23.attention.to_v
  78. - layers.23.attention.to_out.0
  79. - layers.23.feed_forward.w2
  80. - layers.23.feed_forward.w3
  81. - layers.24.attention.to_q
  82. - layers.24.attention.to_k
  83. - layers.24.attention.to_v
  84. - layers.24.attention.to_out.0
  85. - layers.24.feed_forward.w2
  86. - layers.24.feed_forward.w3
  87. - layers.25.attention.to_q
  88. - layers.25.attention.to_k
  89. - layers.25.attention.to_v
  90. - layers.25.attention.to_out.0
  91. - layers.25.feed_forward.w2
  92. - layers.25.feed_forward.w3
  93. - layers.26.attention.to_q
  94. - layers.26.attention.to_k
  95. - layers.26.attention.to_v
  96. - layers.26.attention.to_out.0
  97. - layers.26.feed_forward.w2
  98. - layers.26.feed_forward.w3
  99. - layers.27.attention.to_q
  100. - layers.27.attention.to_k
  101. - layers.27.attention.to_v
  102. - layers.27.attention.to_out.0
  103. - layers.27.feed_forward.w2
  104. - layers.27.feed_forward.w3
  105. - layers.28.attention.to_q
  106. - layers.28.attention.to_k
  107. - layers.28.attention.to_v
  108. - layers.28.attention.to_out.0
  109. - layers.28.feed_forward.w2
  110. - layers.28.feed_forward.w3
  111. - layers.29.attention.to_q
  112. - layers.29.attention.to_k
  113. - layers.29.attention.to_v
  114. - layers.29.attention.to_out.0
  115. - layers.29.feed_forward.w2
  116. - layers.29.feed_forward.w3
  117. save:
  118. dtype: bf16
  119. save_every: 250
  120. max_step_saves_to_keep: 20
  121. save_format: diffusers
  122. push_to_hub: false
  123. datasets:
  124. - folder_path: /home/ai-toolkit/datasets/example
  125. mask_path: null
  126. mask_min_value: 0.1
  127. default_caption: ''
  128. caption_ext: txt
  129. caption_dropout_rate: 0.05
  130. cache_latents_to_disk: false
  131. is_reg: false
  132. network_weight: 1
  133. resolution:
  134. - 768
  135. controls: []
  136. shrink_video_to_frames: true
  137. num_frames: 1
  138. do_i2v: true
  139. flip_x: false
  140. flip_y: false
  141. train:
  142. batch_size: 1
  143. bypass_guidance_embedding: false
  144. steps: 5000
  145. gradient_accumulation: 1
  146. train_unet: true
  147. train_text_encoder: false
  148. gradient_checkpointing: true
  149. noise_scheduler: flowmatch
  150. optimizer: adamw8bit
  151. timestep_type: sigmoid
  152. content_or_style: balanced
  153. optimizer_params:
  154. weight_decay: 0.0001
  155. unload_text_encoder: false
  156. cache_text_embeddings: true
  157. lr: 0.00025
  158. ema_config:
  159. use_ema: false
  160. ema_decay: 0.99
  161. skip_first_sample: false
  162. force_first_sample: false
  163. disable_sampling: false
  164. dtype: bf16
  165. diff_output_preservation: false
  166. diff_output_preservation_multiplier: 1
  167. diff_output_preservation_class: person
  168. switch_boundary_every: 1
  169. loss_type: mse
  170. model:
  171. name_or_path: Tongyi-MAI/Z-Image-Turbo
  172. quantize: false
  173. qtype: qfloat8
  174. quantize_te: false
  175. qtype_te: qfloat8
  176. arch: zimage:turbo
  177. low_vram: false
  178. model_kwargs: {}
  179. layer_offloading: false
  180. layer_offloading_text_encoder_percent: 1
  181. layer_offloading_transformer_percent: 1
  182. assistant_lora_path: ostris/zimage_turbo_training_adapter/zimage_turbo_training_adapter_v2.safetensors
  183. sample:
  184. sampler: flowmatch
  185. sample_every: 250
  186. width: 1024
  187. height: 1024
  188. samples:
  189. - prompt: a beaituful woman with heavy dark makeup is looking at the camera from a close angle, studio lighting.
  190. width: 768
  191. height: 1120
  192. neg: ''
  193. seed: 42
  194. walk_seed: true
  195. guidance_scale: 1
  196. sample_steps: 8
  197. num_frames: 1
  198. fps: 1
  199. logging:
  200. log_every: 1
  201. use_ui_logger: true
  202. meta:
  203. name: Targeted_14_29_
  204. version: '1.0'
Advertisement
Add Comment
Please, Sign In to add comment