RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
1,331 Commits 9 Branches 0 Tags 95 MiB
Commit Graph

272 Commits

Author SHA1 Message Date
Jaret Burkett be3406140b Remove dev indicators 2026-07-24 13:30:30 -06:00
Jaret Burkett 67984754c3 Put more information about model gating and how to solve it 2026-07-24 09:56:02 -06:00
Jaret Burkett ede6f9ecee Major ui speed improvements. Moved file server out of next js app and made it multithreadded. Significantly faster downloads for files, images, and videos. 2026-07-23 09:35:08 -06:00
Jaret Burkett 1086bd0b3e Improvements to file transfer speed when downloading loras from cloud 2026-07-23 08:17:17 -06:00
Jaret Burkett 3a4341dee3 Improve ui polling code to prevent poll buildup 2026-07-21 11:35:52 -06:00
Jaret Burkett e54a0fe78c Move to a single prisma client connection for the ui backend to prevent competing connections. 2026-07-21 11:21:28 -06:00
Jaret Burkett 9e9439015e Switch sqlite database to WAL mode. It will be significantly faster with multiple writers and readers. 2026-07-21 11:07:18 -06:00
fatalis c2864bba48
Enable val/loss by default on the loss graph (#966) 2026-07-20 11:33:03 -06:00
Jaret Burkett 6e158dd1f1 Adjust validation loss defaults 2026-07-19 19:35:33 -06:00
Jaret Burkett 1eb97b7443 Added validation loss 2026-07-19 19:33:45 -06:00
Jaret Burkett cd677c70b5 Fix issue where more than 12 samples would break sample grid. 2026-07-19 09:21:44 -06:00
fatalis 7ba7e35e19
Fix several cases of silent crashing on UI (#922) 
Co-authored-by: Jaret Burkett <jaretburkett@gmail.com>
 2026-07-19 08:23:57 -06:00
Jaret Burkett 988d891102 Added a Sample Next Step in the job gear dropdown to force a sample on the next step. 2026-07-17 07:59:30 -06:00
Jaret Burkett 7a3d94ed03 Add caching to active job pull 2026-07-16 16:51:05 -06:00
Jaret Burkett bf15b65972 Add caching for api calls to speed them up. Added caching for cpu and gpu stat calls 2026-07-16 16:47:10 -06:00
Jaret Burkett 0552d85aa7 Gice the loss graph more diverse colors 2026-07-16 16:06:05 -06:00
Jaret Burkett e1e1996c16 Rework the logging and terminal emulator on the ui to work like an actual emulator for better output. 2026-07-16 08:50:40 -06:00
fatalis 741aeb9ce0
Clear stale return-to-queue flag when starting jobs, fixes crash loop (#920) 2026-07-15 12:48:39 -06:00
rmatif 3e6bd874c4
feat: Add Anima support (#860) 
* Add Anima training support

* Update Anima modular training

* Use sample guidance for Anima

* Fix Anima sampling

* Limit Anima LoRA targets

* Convert Anima LoRA exports

* Fix Anima local loading

* Update Anima default model

* Pin upstream Anima diffusers

* Adjust template defaults to be consistent with other models. Update README

---------

Co-authored-by: Jaret Burkett (Ostris) <jaretburkett@gmail.com>
 2026-07-15 11:59:01 -06:00
fatalis 8bbd051667
Add sample_start_step setting to configure when sampling starts (#949) 
Co-authored-by: Jaret Burkett <jaretburkett@gmail.com>
 2026-07-15 11:15:50 -06:00
Jaret Burkett 691ddf434e Add Qwen3.6 VL captioner. 2026-07-15 07:02:33 -06:00
Jaret Burkett 8b5bf25b13 Add convrot quants to captioner 2026-07-14 07:03:10 -06:00
Jaret Burkett 676b4f3c4c Add Automagic3 optimizer to the ui 2026-07-14 06:39:10 -06:00
Jaret Burkett dcb3b329b2 Fix issue with the graph with sparse data when shrinking smoothing 2026-07-13 17:10:38 -06:00
Jaret Burkett 1f7d608e20 Show sparse datapoints on the loss graph. 2026-07-13 10:59:57 -06:00
Jaret Burkett 28b05ee4ed Added convrotbitnet 1.58 bit quantization 2026-07-13 07:52:48 -06:00
Jaret Burkett a259fa07cd Make convrot ui clearer 2026-07-13 06:44:27 -06:00
Jaret Burkett b6ff367633 Convrot improvements. Add argitrary bit packed byte quantizations. 2026-07-13 06:33:54 -06:00
Jaret Burkett 095d6e7418 Add convrot4 and convrot8 quantizations to the ui. A lot of improvements to the convrot quantization methods. 2026-07-11 13:49:41 -06:00
Jaret Burkett fed9357234 Fix ui to show kv cache for krea2 edit raw 2026-07-09 15:48:46 -06:00
Jaret Burkett a4bbe167ce Added reference token attention isolation (kv_cache) for Krea2 edit training. Same training cost with significant inference speed up. 2x inference speedup. 2026-07-09 12:00:55 -06:00
Jaret Burkett 6233efe1bb Fix sampling issue on zimage turbo and krea2 turbo models 2026-07-08 12:34:39 -06:00
Jaret Burkett 17bc302d13 Make log calls blocking to prevent duplicates. 2026-07-06 07:38:20 -06:00
Jaret Burkett 6c0d1c4679 Improve latency of the job list endpoint 2026-07-04 13:22:09 -06:00
Jaret Burkett 3a94591c89 Disable text encoder unload with krea2 o-edit 2026-07-04 09:17:15 -06:00
Jaret Burkett b1e1a834d4 Added support to train Krea2 as an edit model 2026-07-04 09:12:39 -06:00
Jaret Burkett f63221e577 Replace all sync functions with async to allow more parallel api calls 2026-07-01 17:12:43 -06:00
Jaret Burkett 48781f900b Improve the loading and transfer speed of the dataset file lists 2026-07-01 16:40:01 -06:00
Jaret Burkett b36a8e9c4b Only pull the new log bytes from the ui to improve performance 2026-07-01 14:11:23 -06:00
Jaret Burkett 733e14cb58 Fix issue where the window would scroll to and hilight custom items in the select box 2026-07-01 13:57:56 -06:00
Jaret Burkett 7a089fd0d7 Add support for training directly on Krea2 Turbo with a training adapter 2026-06-23 20:32:02 -06:00
Jaret Burkett (Ostris) 99be3d96a2
Add support for Krea2 (#906) 
* Add support for krea2

* Update repo pointer to actual repo
 2026-06-23 09:18:14 -06:00
Jaret Burkett 548a286992 Add captioning monitoring information to the dataset page so you can just stay on that page with all the info there. 2026-06-19 08:00:40 -06:00
fatalis 4af4fb9d58
Add auto_frame_count support for all remaining video models (#897) 2026-06-19 07:05:18 -06:00
Jaret Burkett 60c1ac6a50 Add support for Boogu Image and Boogu Image Edit 2026-06-18 15:05:49 -06:00
Jaret Burkett e8c828089a Fix issue where gpu sometimes doesnt show on caption modal 2026-06-18 09:37:59 -06:00
Jaret Burkett 178eb5fbbe Add unconditional lora support so Ideogram 4 inference will more closely resemble the full pipeline results. I pushed a finetuned unconditional lora to the hub as an adapter. 2026-06-16 13:27:43 -06:00
Jaret Burkett 86b19589a0 Update the Ideogram 4 prompt generation/parsing/ui to handle the updated format notes better. 2026-06-16 09:44:38 -06:00
Jaret Burkett 6c3b82696e Add support for PRX Pixel T2I 2026-06-13 09:47:53 -06:00
Jaret Burkett 0165fb2ac6 Update npm packages 2026-06-12 17:50:47 -06:00