ai-toolkit/extensions_built_in
Jaret Burkett 139a38f5bd Upcast adaln pruned layers to fp32 on h3 to prevent overflow. 2026-08-06 07:51:15 -06:00
..
advanced_generator Add sample_start_step setting to configure when sampling starts (#949) 2026-07-15 11:15:50 -06:00
audio_models Enable caching of ACE step latents. 2026-04-28 13:39:20 -06:00
captioner Add build support for Nvidia Spark 2026-07-28 12:27:51 -06:00
concept_replacer Add sample_start_step setting to configure when sampling starts (#949) 2026-07-15 11:15:50 -06:00
concept_slider Regularize the slider targets. 2025-09-17 09:36:33 -06:00
dataset_tools Added some split prompting started code, adamw8bit, replacements improving, learnable snr gos. A lot of good stuff. 2023-11-01 06:52:21 -06:00
diffusion_models Upcast adaln pruned layers to fp32 on h3 to prevent overflow. 2026-08-06 07:51:15 -06:00
flex2 Added mask diffirential mask dialation for flex2. Handle video for the i2v adapter 2025-04-10 11:50:01 -06:00
image_reference_slider_trainer Add sample_start_step setting to configure when sampling starts (#949) 2026-07-15 11:15:50 -06:00
sd_trainer Apply sigma to contrastive guidance to balance loss better. Prevent noise grads on images/non audio datasets. Prep for training adapters on MiniMax H3 2026-08-05 13:27:12 -06:00
ultimate_slider_trainer Add sample_start_step setting to configure when sampling starts (#949) 2026-07-15 11:15:50 -06:00