ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity
943 Commits 10 Branches 0 Tags
4d5a649a7dc1e3056e97f66b62e5d32c23d3c8a1
Commit Graph

943 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Jaret Burkett
4d5a649a7d Added initial support for Qwen-Image-2512 2025-12-31 06:11:56 -07:00
Jaret Burkett
0d5c181843 Fixed issue where the control images would sometimes be ignored on qwen_image_edit_2511 2025-12-30 16:40:49 +00:00
Jaret Burkett
356449ec3f bump diffusers version 2025-12-26 11:18:23 -07:00
raziel2001au
90fc99f486 Separate dependencies for DGX OS devices (#610) 2025-12-26 08:32:26 -07:00
Jaret Burkett
a767b82b60 Fixed issue with new logger when ooming 2025-12-25 16:57:34 +00: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
ed36edd85b Bumped docker OS, python version, torch version 2025-12-23 19:44:20 -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
87edca1b2b Added initial support to initiate lora training from an existing lora 2025-12-22 12:49:15 -07:00
raziel2001au
91342853c1 Add support for DGX OS (#567) 2025-12-20 07:20:07 -07:00
Jaret Burkett
8864ba915e Remove easy-dwpose from the default requierments 2025-12-20 07:16:20 -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
Jaret Burkett
3b6c1ade18 Update supporters 2025-12-17 15:20:55 -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
2d30dc5d52 Bump version 2025-12-02 21:29:19 -07:00
Jaret Burkett
6c85184441 Set zit training adapter to default to v2 2025-12-02 16:46:05 -07:00
Jaret Burkett
e6c5aead3b Fix issue that prevented ramtorch layer offloading with z_image 2025-12-02 16:14:34 -07:00
Jaret Burkett
d42f5af2fc Fixed issue with DOP when using Z-Image 2025-11-28 09:36:21 -07:00
Jaret Burkett
08a39754a4 Fixed issue that prevented caching text embeddings on z-image 2025-11-28 09:19:39 -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
21bb8a2bf4 Merge pull request #525 from ostris/flux2 
Add support for FLUX.2
 2025-11-25 07:53:36 -08: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
50e5d99545 Fix issue where text encoder was not fully unloaded in some instances 2025-11-19 09:01:00 -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
323b4aaf5a Do not copy pin memory if it fails, just move 2025-11-17 18:04:00 +00:00
Jaret Burkett
2e7b2d9926 Added Differential Guidance training target 2025-11-10 09:38:25 -07:00
Jaret Burkett
9b89bab8fe Version bump 2025-11-09 10:11:25 -07:00
Jaret Burkett
6f308fc46e When soing guidance loss, make CFG zero an optional target instead of a forced one. 2025-11-04 09:16:15 -07:00
Jaret Burkett
c984369294 Fixed resizing of control image resolution for Qwen Image Edit 2509 when using match_target_res 2025-10-30 06:30:01 -06:00
Jaret Burkett
42e5e3cd1c Adjust DFE to handle 5 dimension latent spaces 2025-10-27 07:48:44 -06:00
Jaret Burkett
8c12977891 Fixed adafactor eps 2025-10-26 05:47:25 -06:00
Jaret Burkett
80418209b8 Fixe a variable that could nt be declared when doing blank prompt preservation 2025-10-23 16:19:56 -06: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
ff14cd6343 Fix check for making sure vae is on the right device. 2025-10-21 14:49:20 -06:00
Jaret Burkett
5123090f6c Adjust dataloader tester to handle videos to test them 2025-10-21 14:47:23 -06:00
Jaret Burkett
0d8a33dc16 Offload ARA with the layer if doing layer offloading. Add support to offload the LoRA. Still needs optimizer support 2025-10-21 06:03:27 -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
8bbaa4e224 Update sponsors 2025-10-20 09:59:44 -06:00
Jaret Burkett
b7f85928f3 Fix issue with chroma when not quantizing 2025-10-19 12:13:05 -06:00
Jaret Burkett
d51297bcf9 Updated supporters in the Readme 2025-10-18 03:14:03 -06:00
Jaret Burkett
1f81bc4060 Fix issue where text encoder could be the wrong quantization and fail when using memory manager 2025-10-15 11:01:30 -06:00
Jaret Burkett
7abf5e20be Add conv3d to memory management excluded modules 2025-10-15 10:12:06 -06:00