ai-toolkit/scripts
Jaret Burkett 78cf049c29
Add support for ACE-Step 1.5 and ACE-Step 1.5 XL. Also added dataset captioning through the UI. (#785)
* Base ace step 1.5 xl added. Generating, still wip on training and ui

* Base training code done

* Fix some issues with caching text embeddings. Update sample cards to show audio

* Fix issue with quantizing ace step

* Add album artwork to samples with waveform.

* Cleanup logs

* Add album art endpoint to speed up album art loading

* Made an make video with artwork script

* Make ui handle basic audio models. Make multi line adjustments to the editor and better syntax hilighting.

* Add prompt tagging system for special tagged models.

* prompt tagging processing for ui working.

* Moved default samples to a special file so we can add more when needed and they can be adjusted for a specific model

* Add a captioner job with music captioner that is prepped for use with the ui

* Add basit ui setup for captioning modal and handeling captioning jobs

* Starting captioning job from ui working. Still better management for it.

* Better filtering of job options in the job view for captioning jobs

* Added qwen3 vl as a captioner for images

* Have an indicator when a dataset is being captioned.

* Adjust the way caption jobs look in the queue

* Fix a few issues. Adjust defaults.

* Version bump

* Added ace step to the readme.
2026-04-09 15:02:03 -06:00
..
calculate_timestep_weighing_flex.py Added new timestep weighing strategy 2025-06-04 01:16:02 -06:00
caption_audio_dataset.py Add support for ACE-Step 1.5 and ACE-Step 1.5 XL. Also added dataset captioning through the UI. (#785) 2026-04-09 15:02:03 -06:00
convert_cog.py Added experimental concept replacer, replicate converter, bucket maker, and other goodies 2023-09-06 18:50:32 -06:00
convert_diffusers_to_comfy.py Added script to convert diffusers model to ComfyUI variant 2025-01-23 21:30:23 -07:00
convert_diffusers_to_comfy_transformer_only.py Remove ip adapter submodule 2025-04-18 09:59:42 -06:00
convert_lora_to_peft_format.py Make a CFG version of flux pipeline 2024-08-05 16:35:53 -06:00
extract_lora_from_flex.py Added a flag to lora extraction script to do a full transformer extraction. 2025-01-24 09:34:13 -07:00
generate_sampler_step_scales.py Tons of bug fixes and improvements to special training. Fixed slider training. 2023-12-09 16:38:10 -07:00
make_diffusers_model.py Bug fixes. allow for random negative prompts 2024-02-21 04:51:52 -07:00
make_lcm_sdxl_model.py added prompt dropout to happen indempendently on each TE 2023-11-14 05:26:51 -07:00
patch_te_adapter.py Bug fixes. allow for random negative prompts 2024-02-21 04:51:52 -07:00
repair_dataset_folder.py Bug fixes. allow for random negative prompts 2024-02-21 04:51:52 -07:00
update_sponsors.py Small tweaks and bug fixes and future proofing 2025-04-05 12:39:45 -06:00