dfde30f231
Fix issue with ltx2 custom te repo path
2026-03-25 09:50:18 -06:00
7f3309b291
Add support for audo frame count so datasets can have varrying length videos. Varous ltx 2.3 VAE optimizations such as removing tiling articacts, and doing frame split encoding to reduce vram on encoding/decoding.
2026-03-24 12:20:09 -06:00
99a4a5887b
Fix Qwen attention mask crash with diffusers >=0.37 ( #748 )
...
* Fix Qwen Image mask handling
* Fix Qwen attention mask crash with diffusers >=0.37
diffusers v0.37 (PR #12987 ) optimizes all-ones attention masks to None
in encode_prompt() when there is no padding. This breaks ai-toolkit's
Qwen extensions which call .to() on the mask unconditionally.
Fix: reconstruct the all-ones mask at the boundary (get_prompt_embeds)
right after encode_prompt() returns. This keeps the rest of the code
unchanged and works with both old and new diffusers versions.
Also removes redundant duplicate mask assignments in qwen_image_edit
and qwen_image_edit_plus.
Fixes #740
2026-03-23 14:43:08 -06:00
295094b4b5
Fixed new breaking change in diffusers with with qwen image
2026-03-23 14:10:55 -06:00
5642b656b9
Fix audio issues with ltx2 models. Silent codec fails now raised. Auto convert surround sound audio to stereo. Invalidate old caches just to be safe so they recache now.
2026-03-23 20:08:33 +00:00
561e6f201c
Fixed an issue with ltx 2.3 i2v training
2026-03-23 12:41:18 -06:00
e91827f9be
Change gemma repo to lightricks one that is not gated
2026-03-23 11:00:32 -06:00
253cb31362
Fix issue with video and images with no audio on ltx models
2026-03-22 22:09:23 -06:00
4a3d317e2b
Fix issue with using the default text encoder with ltx 2.3
2026-03-22 18:53:59 -06:00
859635e95b
Add support for training LTX 2.3 ( #745 )
...
* Initial support for ltx 2.3. Still needs a lot of testing to make sure it is all right.
* bump version
* Handle lora renaming keys for new ltx 2.3 layers
2026-03-22 17:56:59 -06:00
57d407cfd4
Add support for training lodestones/Zeta-Chroma
2026-03-01 12:52:29 -07:00
e82cf6eec2
Fixed issue that prevented full fine-tuning of flux2 models when using gradient checkpointing
2026-02-06 16:18:43 -07:00
1ce2428722
Shrink text embeds to max token length for LTX-2. Drastically reduces cached text embedding sizes
2026-01-28 12:54:49 -07:00
a6da9e37ac
Add support for FLUX.2 klein base models
2026-01-17 17:46:25 -07:00
0efed794b4
Fix issue where flux2 would ignore single control image on training
2026-01-17 20:26:35 +00:00
e40d7ac605
Ignore i2v on ltx is training on images
2026-01-14 18:46:27 -07:00
9848de7946
Fix issue with ltx cached latents if there is no audio.
2026-01-14 17:27:01 -07:00
73dedbf662
Do caching of latents, first frame and audio when caching latents for LTX2
2026-01-14 11:05:23 -07:00
64fe29b182
Support img 2 vid training for ltx-2
2026-01-13 19:04:56 -07:00
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
0d5c181843
Fixed issue where the control images would sometimes be ignored on qwen_image_edit_2511
2025-12-30 16:40:49 +00:00
e6c5aead3b
Fix issue that prevented ramtorch layer offloading with z_image
2025-12-02 16:14:34 -07:00
4e62c38df5
Add support for training Z-Image Turbo with a de-distill training adapter
2025-11-28 08:08:53 -07:00
01cf480233
Add FLUX.2 official weights
2025-11-25 08:52:19 -07:00
dadbeda197
Update test weights
2025-11-23 10:51:50 -07:00
af8e9ea149
Add initial support for FLUX.2
2025-11-18 11:17:38 -07:00
c984369294
Fixed resizing of control image resolution for Qwen Image Edit 2509 when using match_target_res
2025-10-30 06:30:01 -06:00
ff14cd6343
Fix check for making sure vae is on the right device.
2025-10-21 14:49:20 -06:00
76ce757e0c
Added initial support for layer offloading wit Wan 2.2 14B models.
2025-10-20 14:54:30 -06:00
b7f85928f3
Fix issue with chroma when not quantizing
2025-10-19 12:13:05 -06:00
0c9e1c3deb
Fixed some fringe cases for qwen image edit.
2025-10-13 17:10:46 +00:00
e9c4d94256
Allow for matching target resolution with control images for Qwen Image Edit 2509
2025-10-10 14:24:27 -06:00
1bc6dee127
Change auto_memory to be layer_offloading and allow you to set the amount to unload
2025-10-10 13:12:32 -06:00
8068755b0a
Fixed issue with wan 2.2 getting stuck on CPU
2025-10-09 17:24:25 +00:00
4e5707854f
Initial support for RamTorch. Still a WIP
2025-10-05 13:03:26 -06:00
b7c04efb44
A commit with the adits properly named improvements to qwen image edit plus workflow. Fixed a bug. Dont norm the cfg
2025-10-01 14:13:15 -06:00
3086a58e5b
git status
2025-10-01 14:12:17 -06:00
67ed563e03
fix issue with multi batch size on qwen-image-edit-plus
2025-09-30 09:04:56 -06:00
6da417261c
Add extra detachments just to be sure on qiep
2025-09-27 08:53:59 -06:00
e04f55c553
Fixed scaling issue with control images
2025-09-26 11:49:53 -06:00
454be0958a
Initial support for qwen image edit plus
2025-09-24 11:39:10 -06:00
20dfe1b4d5
Small double tap of detach on qwen just for good measure
2025-09-18 08:22:04 -06:00
b95c17dc17
Add initial support for chroma radiance
2025-09-10 08:41:05 -06:00
056711d4ed
Fix issue with wan22 14b that woudl load both transformers temporarily resulting in oom on 24GB.
2025-08-28 13:06:31 -06:00
119653c3f2
Force width, height, and num frames to always be the proper sizes for Wan 2.2 models
2025-08-25 10:33:28 -06:00
e1fd411665
Added support for Chroma1 official release. Will still use single file verstion instead of the diffusers version.
2025-08-23 09:06:28 -06:00
bf2700f7be
Initial support for finetuning qwen image. Will only work with caching for now, need to add controls everywhere.
2025-08-21 16:41:17 -06:00
83deaec417
Minor bug fixes
2025-08-21 08:05:34 -06:00
d2bbe1872c
Add support for fine tuning Wan 2.2 I2V 14B
2025-08-18 11:43:32 -06:00
b3e666daf4
Fix issue with wan22 14b where timesteps were generated not in the current boundary.
2025-08-16 21:16:48 -06:00
Jaret Burkett
Rayane