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

254 Commits

Author SHA1 Message Date
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
Jaret Burkett
d83f7dd4d9 Fix a few issues with compile. Changed defaults. Future proofed block layer compile. 2026-06-12 11:43:43 -06:00
Jaret Burkett
a5558ae7d9 Add compile to the actual right config section. 2026-06-12 11:12:41 -06:00
Jaret Burkett
c09b228a35 Add model compiling to the ui 2026-06-12 11:02:10 -06:00
Rainer
324faf17b3 accept .json uploads (#880) 2026-06-12 10:31:19 -06:00
Jaret Burkett
0f580f0663 Add compiling option for captioning. Added progress bar for captioning on active job widget. Track steps on caption. Added reverse proxy iframe that plugins can use to add ui functionality. 2026-06-12 10:26:52 -06:00
Jaret Burkett
9cf34f945c Allow multi selecting jobs and deleting them in one go 2026-06-12 08:01:56 -06:00
fatalis
53ebb93edb fix learning rate metric truncating to zero on graph (#877) 2026-06-12 07:30:21 -06:00
Jaret Burkett
9e99d3ce5d Save loss graph view settings per page to local storage. 2026-06-09 07:29:51 -06:00
Jaret Burkett
0d018db689 Rework the toggles so we can hide the trend line from the loss graph. No need for a raw toggle either. 2026-06-08 13:56:37 -06:00
Jaret Burkett
687def6f7a Add better smoothing with ema rounding of the ends of the loss graph for first and last instance do not dominate the smoothing factors. 2026-06-08 10:43:46 -06:00
Jaret Burkett
d7f8887bbf Show all logged metrics on the loss graph 2026-06-08 09:43:03 -06:00
Jaret Burkett
41157b460c Added ability to set the caption extention in dataset viewer, captioner, and trainer so one dataset can have multiple caption styles in different files with different extensions. Added dataset caption template for a blank ideogram 4 formatted template. 2026-06-06 08:32:24 -06:00
Jaret Burkett
21a6beb194 Allow editing the boxes visually on the sample section for ideogram 2026-06-05 14:09:18 -06:00
Jaret Burkett
4441080c05 Rework the ui of ideogram dataset image caption editor 2026-06-05 13:20:58 -06:00
Jaret Burkett
b70083a74f Fix issue with default sampels. 2026-06-05 12:33:42 -06:00
Jaret Burkett
c994398850 Rework prompts and captioning systems for ideogram to more strictly match the format provided by ideogram. 2026-06-05 11:14:36 -06:00
Jaret Burkett
6fd2253932 Imporved editing boxed on the dataset viewer 2026-06-05 10:30:06 -06:00
Jaret Burkett
ef12260b80 Add a prompt upsample ui for upsampling prompts to ideogram format prompts. 2026-06-05 09:56:02 -06:00