github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
1,440 Commits 9 Branches 0 Tags
5f53ecde541eb4bf97ee51300bc9a651f1255ee3
Commit Graph

291 Commits

Author SHA1 Message Date
Jaret Burkett
5f53ecde54 Make the viewer thumbnails pul the thumbs 2026-08-14 18:14:12 -06:00
Jaret Burkett
247cb45c3e Improve gpu cpu monitor efficiency. 2026-08-14 17:14:00 -06:00
Jaret Burkett
695b0baccf Switch to a live always active gpu and cpu device monitor that can be connected to with an SSE connection for real time device information stream. Much more efficient and faster than previous method of polling. 2026-08-14 12:40:01 -06:00
Jaret Burkett
6ea281973d Add support for MiniMax H3 Ref2Vid training 2026-08-13 12:36:03 -06:00
Jaret Burkett
ab18528fdb Let omni captioner handle just images as well. Set it as the new default captioner. 2026-08-13 10:30:17 -06:00
Jaret Burkett
4e91fb2d0a Add thinking and abliterated versions of qwen omni. 2026-08-13 08:56:31 -06:00
Jaret Burkett
6c88e3d138 Add layer offloading to omni captioner 2026-08-13 07:21:54 -06:00
Jaret Burkett
742a4c8cef Added a Download Full Dataset and Download Captions option to the datasets page so they captions or dataset can be easily downloaded. 2026-08-12 21:37:35 -06:00
Jaret Burkett
81adcc2176 Add caption prompt template picker to the ui. A few styling improvements thrown in for a good time. 2026-08-12 21:23:28 -06:00
Jaret Burkett
7f9a142dfd Generate thumbnails for dataset items so things load faster and the ui is more stable. Especially good for videos. 2026-08-12 20:33:42 -06:00
Jaret Burkett
175cc1e151 Add Qwen 3 Omni for captioning videos with sound. 2026-08-12 19:57:01 -06:00
Jaret Burkett
e16e04f123 On the ui, keep the last caption job bar visible at the bottom of a dataset so it can be viewed and edited 2026-08-12 17:32:48 -06:00
Jaret Burkett
18645d93b7 Fix trailing slashes in path for settings 2026-08-12 17:04:22 -06:00
Jaret Burkett
7eb65b837a Switch MiniMax H3 to default to contrastive guidance loss. Add a custom model select toggle to automatically fill out settings for the model. 2026-08-12 07:54:57 -06:00
Jaret Burkett
cbf910ac02 Add support for LTX 2.5 2026-08-12 05:50:30 -06:00
Jaret Burkett
ab5fef8970 Fix caption dropout so it now works identically when caching text embeddings. 2026-08-10 18:53:33 -06:00
Jaret Burkett
5ff8a0435a Make v1 the default training adapter 2026-08-09 14:56:39 -06:00
Jaret Burkett
9171d5ec1d Update training adapter path 2026-08-06 11:21:28 -06:00
Jaret Burkett
71625d1207 Add the alpha version of the MiniMax H3 trianing adapter and set it as the new default training method. 2026-08-06 11:11:31 -06:00
Jaret Burkett
1e1418b22c Apply sigma to contrastive guidance to balance loss better. Prevent noise grads on images/non audio datasets. Prep for training adapters on MiniMax H3 2026-08-05 13:27:12 -06:00
Jaret Burkett
9fc1f208df Remove adaln_proj from the network modules for minimax_h3 2026-08-04 16:19:46 -06:00
Jaret Burkett
183433ae8e Make MiniMax H3 default to using contrastive guidance loss to prevent distillation breakdown 2026-08-04 16:13:45 -06:00
Jaret Burkett
18f5810d6c Adjust default alpha for h3 2026-08-03 20:15:28 -06:00
Jaret Burkett
88ac27fc8f Handle images with MiniMax H3. 2026-08-03 11:58:44 -06:00
Jaret Burkett
8502a845b1 Add support for MiniMax H3 T2V and I2V training 2026-08-03 10:17:39 -06:00
Jaret Burkett
038270eb2f Fix issue with macstats on mac 2026-07-28 08:27:31 -06:00
Jaret Burkett
461e798708 Rework windows start and stop methods so that command windows dont appear. Stop with signal since we cannot signint. 2026-07-27 17:43:01 -06:00
Jaret Burkett
92bdb6e473 Add support for Mage-Flow and Mage-Flow Edit 2026-07-25 11:34:20 -06:00
Jaret Burkett
efb58c8641 Switch to thumbnails and thumbnail creation on the sample grid page until clicked. 2026-07-25 08:40:33 -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
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
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
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
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