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
361 Commits 10 Branches 0 Tags
8d48ad4e85874b3dca610a659e3e70233bc38ebc
Commit Graph

20 Commits

Author SHA1 Message Date
Jaret Burkett
8d48ad4e85 fixed bug I added to demo config 2024-08-11 10:28:39 -06:00
Jaret Burkett
ec1ea7aa0e Added support for training on primary gpu with low_vram flag. Updated example script to remove creepy horse sample at that seed 2024-08-11 09:54:30 -06:00
Jaret Burkett
2308ef2868 Added flux training instructions 2024-08-10 14:10:02 -06:00
Jaret Burkett
c6675e2801 Added shuffeling to prompts 2023-08-19 07:57:30 -06:00
Jaret Burkett
90eedb78bf Added multiplier jitter, min_snr, ability to choose sdxl encoders to use, shuffle generator, and other fun 2023-08-19 05:54:22 -06:00
Jaret Burkett
8c90fa86c6 Complete reqork of how slider training works and optimized it to hell. Can run entire algorythm in 1 batch now with less VRAM consumption than a quarter of it used to take 2023-08-05 18:46:08 -06:00
Jaret Burkett
66c6f0f6f7 Big refactor of SD runner and added image generator 2023-08-03 14:51:25 -06:00
Jaret Burkett
2bf3e529ce Set gradient checkpointing on unet enabled by default. Help out immensly with sdxl backprop spikes 2023-08-01 15:43:27 -06:00
Jaret Burkett
8b8d53888d Added Model rescale and prepared a release upgrade 2023-08-01 13:49:54 -06:00
Jaret Burkett
61dd818608 Added anchors to regulate the lora 2023-07-24 14:59:16 -06:00
Jaret Burkett
9a2819900c added target weight to targets 2023-07-23 14:08:37 -06:00
Jaret Burkett
452f2a6da2 Added info, config, etc for lora extracotr and slider trainer 2023-07-23 13:13:45 -06:00
Jaret Burkett
9367089d48 Added example for slider training that will run as is 2023-07-23 11:24:12 -06:00
Jaret Burkett
982e0be7a9 Removed train config, updating it, and added my llvae as pytorch model 2023-07-20 14:12:28 -06:00
Jaret Burkett
78b59c5e99 Added support for 3cleir, not fully tested 2023-07-16 15:35:14 -06:00
Jaret Burkett
8d6edae9fd Added support for traditional LoRa extract using LoCon script 2023-07-12 19:51:40 -06:00
Jaret Burkett
57f14e5ef2 WIP implementing training 2023-07-12 08:23:46 -06:00
Jaret Burkett
47d094e528 Setup base for training jobs. Added sd-scripts as a submodule 2023-07-08 13:50:59 -06:00
Jaret Burkett
37354b006e Reworked so everything is in classes for easy expansion. Single entry point for all config files now. 2023-07-08 09:51:42 -06:00
Jaret Burkett
e4de8983c9 Initial commit 2023-07-05 16:44:58 -06:00