ai-toolkit

mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00

Files

Jaret Burkett 5b5aadadb8 Add LTX-2 Support (#644 )

* WIP, adding support for LTX2

* Training on images working

* Fix loading comfy models

* Handle converting and deconverting lora so it matches original format

* Reworked ui to habdle ltx and propert dataset default overwriting.

* Update the way lokr saves to it is more compatable with comfy

* Audio loading and synchronization/resampling is working

* Add audio to training. Does it work? Maybe, still testing.

* Fixed fps default issue for sound

* Have ui set fps for accurate audio mapping on ltx

* Added audio procession options to the ui for ltx

* Clean up requirements

2026-01-13 04:55:30 -07:00

compare_keys.py

Added method to get specific keys from model

2023-08-15 14:51:04 -06:00

generate_lora_mapping.py

Corrected key saving and loading to better match kohya

2023-09-04 00:22:34 -06:00

generate_weight_mappings.py

More guidance work. Improved LoRA module resolver for unet. Added vega mappings and LoRA training for it. Various other bigfixes and changes

2023-12-15 06:02:10 -07:00

merge_in_text_encoder_adapter.py

Added caching to image sizes so we dont do it every time.