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

1358 Commits

Author SHA1 Message Date
Jaret Burkett fe82487187 Add tiling on vae decode for qwen image models when low vram flag is on 2026-07-10 09:07:29 -06:00
Jaret Burkett e7951ad29e Exclude sensative layers from quantization on wan models 2026-07-10 08:56:05 -06:00
Jaret Burkett 883d60eb71 Do vae tiling when decoding wan models with low vram active. 2026-07-10 08:02:32 -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 5a9b5bde3f Added experimental orbit quant 2026-07-09 15:14:51 -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 dd08579eda Add ability to pull control images from same folder group 2026-07-08 05:21:31 -06:00
Jaret Burkett 7bceec3b07 Apply a loss mask for first frame conditioning for wan and ltx 2026-07-07 09:05:53 -06:00
Jaret Burkett bd93a312bc Load more types of comfy ui style quants for ltx2 2026-07-07 09:04:51 -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 4e50535478 Allow models to provide an additional loss 2026-06-28 09:49:36 -06:00
Jaret Burkett 1e12b6b73f Remove duplicate save when triggering save next step 2026-06-28 07:18:16 -06:00
Jaret Burkett 7ee1f98f6d Allow loading and saving z_image in comfy safetensors format. 2026-06-28 07:15:49 -06:00
Jaret Burkett c97fc9973a Fix loading pretraied lora when merging network on save 2026-06-28 06:04:36 -06:00
Jaret Burkett f8667f0334 Save Krea2 even if quantized 2026-06-28 06:03:37 -06:00
Jaret Burkett df6ea4263d Version bump 2026-06-26 13:02:10 -06:00
Jaret Burkett ad87aacec0 Add ability to set certain layers to full for loras 2026-06-26 13:01:48 -06:00
Jaret Burkett 4a99ddabad Fix breaking change with diffusers qwen image 2026-06-26 10:04:35 -06:00
Jaret Burkett 5f04ae7ad5 Rework merge_network_on_save to handle dequantization on merging and saving for much more efficient full finetuning. 2026-06-25 13:19:38 -06:00
Jaret Burkett 6ecff36f26 Add a way to do full rank lora modules of non lora layers such as embeddings, norm, etc 2026-06-25 11:55:07 -06:00
Jaret Burkett 4eb0707639 Add a control generation script. 2026-06-25 10:14:24 -06:00
Jaret Burkett d14f6e567a Allow individual models to scale the loss after it is calculated. 2026-06-25 10:13:58 -06:00
Jaret Burkett f743ccf7ef Save before last sample 2026-06-24 09:13:01 -06:00
PlagueKind 089e41dd1c
Compile improvements - auto cache size, fix fullgraph setting, fix triton detection (#899) 
* Compile improvements - auto cache size, fix fullgraph setting, fix triton detection

* remove forced torchao no longer needed
 2026-06-24 07:59:41 -06:00
fatalis d586125b40
Force HF downloads to show progress bars when output is not a TTY (#909) 2026-06-24 07:55:38 -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 a803611ec1 Enable tiling on vae when decoding with low_vram flag on krea2 2026-06-23 18:38:52 -06:00
Jaret Burkett 724e67d634 Set krea 2 to use new lokr format 2026-06-23 13:31:50 -06:00
Jaret Burkett e20b42e84a Add offloading support for krea2 2026-06-23 10:43:36 -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 af594061ab Add ability to do hidden states with tipsv2 2026-06-22 11:00:04 -06:00
Jaret Burkett 820d534d6e Add features for models that may need a non masked loss such as inpainiting. 2026-06-22 10:59:17 -06:00
Jaret Burkett c133c55cf5 Fix issue with mask generator revision 2026-06-22 08:19:27 -06:00
Jaret Burkett d51463ca52 Add ability to recover from a truncated image file in the dataset 2026-06-22 05:07:30 -06:00
Jaret Burkett ba0b3dbb65 Force batch size when bucket is too small by duplicating items in the batch 2026-06-21 20:01:29 -06:00
Jaret Burkett dba092fc15 Keep a real git repo in the docker image 2026-06-21 06:54:19 -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
Jaret Burkett 99f8fd44e3 Add a fallback bbox adjustment when json parsing fails on ideogram4 captioner 2026-06-19 07:25:54 -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 022d1c29e0 Remove color pallet for individual objects in the caption prompts 2026-06-19 06:56: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 e886745051 Added additional information on addine new models and some additional gotchas 2026-06-18 15:04:50 -06:00