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
26 Commits 10 Branches 0 Tags
9367089d488048f8fde982f7fca089b33d07753a
Commit Graph

26 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Jaret Burkett
9367089d48 Added example for slider training that will run as is 2023-07-23 11:24:12 -06:00
Jaret Burkett
434fb22458 Adde dnoise offset 2023-07-22 15:01:01 -06:00
Jaret Burkett
3f4f429c4a Reworked the balancing and swapping of the lora during training to make it much more stable when trained 2023-07-22 14:13:39 -06:00
Jaret Burkett
ce8e7a1271 Added prompt flag to adjust network multiplier 2023-07-22 00:05:15 -06:00
Jaret Burkett
596e59dd6d Slider training functioning, time to perfect it 2023-07-21 22:06:49 -06:00
Jaret Burkett
ddcd9069e1 Base for loopback lora training setup, still working on proper sliders 2023-07-21 18:26:02 -06:00
Jaret Burkett
aa13251877 Actually addind the Process class..... 2023-07-20 15:46:23 -06:00
Jaret Burkett
0761656a90 Added my good ole pattern loss. God I love that thing, conv transpose pattern instantly wiped from vae 2023-07-20 15:44:16 -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
c29b9d075f Setup the base process for merging things. WIP 2023-07-20 07:39:31 -06:00
Jaret Burkett
557732e7ff Added Critic support to VAE training. Still tweaking and working on it. Many other fixes 2023-07-19 15:57:32 -06:00
Jaret Burkett
6ada328d8d Fixed issues with vgg19 preprocessing. Added yaml support on config file 2023-07-19 06:12:57 -06:00
Jaret Burkett
8d8229dfc0 Handle conversions back to ldm for saving 2023-07-18 19:34:35 -06:00
Jaret Burkett
17c13eef88 Default to only training mse. Did a lot of cleanup with script. Added logging via tensorboard. 2023-07-18 09:40:51 -06:00
Jaret Burkett
94d52572d4 Style and content loss working 2023-07-18 07:47:01 -06:00
Jaret Burkett
439310e4dc Made a very basic vae trainer. 2023-07-17 19:03:50 -06:00
Jaret Burkett
78b59c5e99 Added support for 3cleir, not fully tested 2023-07-16 15:35:14 -06:00
Jaret Burkett
9df7af1720 Adjusted some names and allow to ignore text encoder on extract 2023-07-16 12:52:39 -06:00
Jaret Burkett
a61914ddd9 Add support for threshold, ratio, and quantile on traditional LoRa 2023-07-12 20:02:22 -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
27df03a486 Actually fixed meta data 2023-07-05 22:00:58 -06:00
Jaret Burkett
18077ba265 Fixed issue with meta. 2023-07-05 16:49:59 -06:00
Jaret Burkett
e4de8983c9 Initial commit 2023-07-05 16:44:58 -06:00