evan/ai-toolkit
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
1,614 Commits 2 Branches 0 Tags
577004a9cd281035f3913f0db7ef34616e04b6e1
Commit Graph
353 Commits
Author SHA1 Message Date
Rayane ca72eb1515 Add 1328 native resolution for Qwen Image training (#749) 
* Add 1328 native resolution for Qwen Image training

Qwen-Image and Qwen-Image-2512 have a native 1:1 resolution of 1328x1328
as documented in the official model card's aspect ratio table. Adding it
to the resolution buckets and UI allows training at the model's native
resolution for improved quality.

* Revert example config change (24GB OOM at 1328)
 2026-03-28 12:09:15 -06:00
Jaret Burkett 740657e25e Improve dataset uploader. Upload the files one at a time instead of one huge chunk. Show progress for each file. 2026-03-27 09:26:22 -06:00
Jaret Burkett 2c8d2acdcb On jobs table, sort idle jobs by last updated so recent active ones are at the top 2026-03-26 10:33:17 -06:00
Jaret Burkett 9a77389653 On a new training job, or when editing one, load everything before allowing editing 2026-03-26 10:23:42 -06:00
Jaret Burkett a7bb4ddb2c Work on loss graph. Add smoothed overlay. Allow user to hilite a secton of the graph to zoom into. 2026-03-26 10:05:09 -06:00
Jaret Burkett 4df3b0463f Save job pid to the database and sing sigint to kill it when stopping so it stops immediatly. 2026-03-26 09:10:37 -06:00
Jaret Burkett 89d2090962 Fixed race condition that would occasionally set the dataset path to the first one when editing a job 2026-03-25 11:22:42 -06:00
Jaret Burkett 5d5a8ef9da Fixed issue with deleting datasets and jobs with newer version of node.js. Bumped minimum version of node js to 20 2026-03-25 10:04:36 -06:00
Jaret Burkett 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
Rasmus Lerdorf 4ad14d211a Add an import config button (#733) 2026-03-23 15:41:27 -06:00
Jaret Burkett 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
Jaret Burkett 35978df8a3 Adjust defaults for ui graph to get and show all losses 2026-03-02 10:27:15 -07:00
Jaret Burkett 57d407cfd4 Add support for training lodestones/Zeta-Chroma 2026-03-01 12:52:29 -07:00
Jaret Burkett 1c74ca5d22 Add audio_loss_multiplier to scale audio loss to larger values if desired. 2026-02-19 11:57:44 -07:00
Jaret Burkett 42acb0d4be Build out an audio player card in preperation for audio datasets and samples. 2026-02-03 08:15:55 -07:00
Jaret Burkett ea912d2d7b Increase default sample steps from 25 to 30 for z_image 2026-01-27 09:39:21 -07:00
Jaret Burkett 2db090144a Add support for Z-Image 2026-01-27 09:34:46 -07:00
Jaret Burkett 9ef6f1a828 Increase client body size to 100 gb 2026-01-24 12:44:17 -07:00
Jaret Burkett a6da9e37ac Add support for FLUX.2 klein base models 2026-01-17 17:46:25 -07:00
Jaret Burkett e132dbae76 Add number of repeats for a dataset in the ui 2026-01-15 08:03:31 -07:00
Jaret Burkett 64fe29b182 Support img 2 vid training for ltx-2 2026-01-13 19:04:56 -07:00
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
Jaret Burkett 6870ab490f Add 4 bit ARA for qwen image 2512 2026-01-01 20:16:56 -07:00
Jaret Burkett 926097aa4c Added 3bit ARA for qwen image 2512 2026-01-01 06:51:49 -07:00
Jaret Burkett 4d5a649a7d Added initial support for Qwen-Image-2512 2025-12-31 06:11:56 -07:00
Jaret Burkett 8edf1e44c5 Added 3 bit accuracy recovery adapter for qwen image edit 2511 2025-12-24 05:33:47 -07:00
Jaret Burkett 57a2ab1299 Add initial support for Qwen Image Edit 2511 2025-12-23 10:53:48 -07:00
Jaret Burkett 9883055684 Fix issue where ui could break if caption is read as a non string. 2025-12-23 09:47:19 -07:00
Jaret Burkett 113bbd0e3e Fixed an issue with the new version of nextjs with client body size 2025-12-18 19:18:35 -07:00
Jaret Burkett ba00eea7d9 Add loss graph to the ui 2025-12-18 10:08:59 -07:00
apolinário cd0e691040 Fix NextJS vulnerability (#594) 
* Fix NextJS vulnerability

https://nextjs.org/blog/CVE-2025-66478

* Update package-lock.json

* Update package-lock.json

* Update package.json

* Update package-lock.json
 2025-12-17 10:03:13 -07:00
Jaret Burkett 26f4f02453 Add support for Z-Image-De-Turbo 2025-12-04 10:03:13 -07:00
Jaret Burkett 6c85184441 Set zit training adapter to default to v2 2025-12-02 16:46:05 -07:00
Jaret Burkett 4e62c38df5 Add support for training Z-Image Turbo with a de-distill training adapter 2025-11-28 08:08:53 -07:00
Jaret Burkett 01cf480233 Add FLUX.2 official weights 2025-11-25 08:52:19 -07:00
Jaret Burkett dadbeda197 Update test weights 2025-11-23 10:51:50 -07:00
Jaret Burkett 0b5f3475e2 Merge branch 'main' into flux2 2025-11-23 08:18:35 -07:00
Jaret Burkett 26e4b71b57 Fix issue with parsing image info for sample info on some windows machines. 2025-11-19 08:45:05 -07:00
Jaret Burkett cd607c4902 Fix a bug that can happen if you remove a gpu from your machine. 2025-11-19 08:28:48 -07:00
Jaret Burkett af8e9ea149 Add initial support for FLUX.2 2025-11-18 11:17:38 -07:00
Jaret Burkett 2e7b2d9926 Added Differential Guidance training target 2025-11-10 09:38:25 -07:00
Jaret Burkett ee206cfa18 Added blank prompt preservation 2025-10-22 14:55:13 -06:00
Jaret Burkett ca57ffc270 When having less than 3 sample images, add spacing to the grid so images are not huge 2025-10-22 13:52:38 -06:00
Jaret Burkett 76ce757e0c Added initial support for layer offloading wit Wan 2.2 14B models. 2025-10-20 14:54:30 -06:00
Jaret Burkett 645c54d617 Fixed issue that may occour if no queue is built when starting one from the table. 2025-10-15 08:48:07 -06:00
Jaret Burkett b523d58699 Added ability to clone an existing job in the ui 2025-10-14 14:13:37 -06:00
Jaret Burkett 7e34a03113 Added queing system to the UI 2025-10-14 12:00:42 -06:00
Jaret Burkett e9c4d94256 Allow for matching target resolution with control images for Qwen Image Edit 2509 2025-10-10 14:24:27 -06:00
Jaret Burkett 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
Jaret Burkett dfc85f0b51 Add Auto Memory for qwen models in the ui 2025-10-07 10:43:52 -06:00