Jaret Burkett
|
2cb27c3f57
|
Merge branch 'main' of github.com:ostris/ai-toolkit
|
2023-08-26 08:55:09 -06:00 |
|
Jaret Burkett
|
3367ab6b2c
|
Moved SD batch processing to a shared method and added it for use in slider training. Still testing if it affects quality over sampling
|
2023-08-26 08:55:00 -06:00 |
|
Jaret Burkett
|
24f46ea7d6
|
Merge pull request #9 from FoundSol/patch-1
Update README.md
|
2023-08-25 20:05:51 -06:00 |
|
Fundamentum
|
5bef2985b5
|
Update README.md
|
2023-08-25 21:13:24 -03:00 |
|
Jaret Burkett
|
aeaca13d69
|
Fixed issue with shuffeling permutations
|
2023-08-23 22:02:00 -06:00 |
|
Jaret Burkett
|
b408f9f3eb
|
Fixed issue with timestep I broke for sliders
|
2023-08-23 16:15:30 -06:00 |
|
Jaret Burkett
|
7157c316af
|
Added support for training lora, dreambooth, and fine tuning. Still need testing and docs
|
2023-08-23 15:37:00 -06:00 |
|
Jaret Burkett
|
e2c547f6c2
|
Fixed typo
|
2023-08-23 13:33:48 -06:00 |
|
Jaret Burkett
|
7b770bc305
|
Merge pull request #7 from ostris/textual_inversion
Textual inversion training
|
2023-08-23 13:31:37 -06:00 |
|
Jaret Burkett
|
f200cf36c5
|
Added train example to ti
|
2023-08-23 13:30:29 -06:00 |
|
Jaret Burkett
|
d298240cec
|
Tied in ant tested TI script
|
2023-08-23 13:26:28 -06:00 |
|
Jaret Burkett
|
2e6c55c720
|
WIP creating textual inversion training script
|
2023-08-22 21:02:38 -06:00 |
|
Jaret Burkett
|
36ba08d3fa
|
Added a converter back to ldm from diffusers for sdxl. Can finally get to training it properly
|
2023-08-21 16:22:01 -06:00 |
|
Jaret Burkett
|
e8667f856f
|
Fix issue with there being an extra . on gene
|
2023-08-20 15:54:38 -06:00 |
|
Jaret Burkett
|
bef5551ea5
|
Ultimate slider training built, still needs tuning
|
2023-08-19 18:54:34 -06:00 |
|
Jaret Burkett
|
b77b9acc0b
|
Added base for ultimate slider. WIP
|
2023-08-19 15:35:24 -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
|
80e2f4a2a4
|
Merge branch 'main' of github.com:ostris/ai-toolkit
|
2023-08-18 11:45:00 -06:00 |
|
Jaret Burkett
|
d51c4ca704
|
Added ability to use two seperate folders for datasets when doing image reference sliders
|
2023-08-18 11:44:33 -06:00 |
|
Jaret Burkett
|
c7ec132d5d
|
third times a charm
|
2023-08-16 20:54:00 -06:00 |
|
Jaret Burkett
|
ed9607e8da
|
Update SliderTraining.ipynb
I should be doing this the right way
|
2023-08-16 20:49:09 -06:00 |
|
Jaret Burkett
|
d44f8ac508
|
Update SliderTraining.ipynb
fixed code block
|
2023-08-16 20:47:56 -06:00 |
|
Jaret Burkett
|
8d09eb44ec
|
Fixed an issue with CFG time embeds on SDXL
|
2023-08-15 18:02:14 -06:00 |
|
Jaret Burkett
|
55a5fcc7d9
|
Added method to get specific keys from model
|
2023-08-15 14:51:04 -06:00 |
|
Jaret Burkett
|
e96874241d
|
Added slider colab to readme
|
2023-08-13 13:55:38 -06:00 |
|
Jaret Burkett
|
e3be1a1758
|
Added WIP slider training colab
|
2023-08-13 13:52:38 -06:00 |
|
Jaret Burkett
|
1a92e97c6d
|
Added missing deps
|
2023-08-13 13:15:04 -06:00 |
|
Jaret Burkett
|
355c80df07
|
Added ability to use civit ai url ar model name and built a model downloader and cache manager for it
|
2023-08-13 13:09:51 -06:00 |
|
Jaret Burkett
|
1487d13191
|
Moved the run job command
|
2023-08-13 10:25:56 -06:00 |
|
Jaret Burkett
|
383bad958d
|
Added a way to run as a library by passing job dict
|
2023-08-13 09:54:39 -06:00 |
|
Jaret Burkett
|
196b693cf0
|
Worked on reference slider script. It is working well currently. Still going to tune it a bit before a writeup though
|
2023-08-12 17:59:24 -06:00 |
|
Jaret Burkett
|
fd95e7b60c
|
Merge branch 'main' of github.com:ostris/ai-toolkit
|
2023-08-12 05:59:58 -06:00 |
|
Jaret Burkett
|
379992d89e
|
Various bug fixes and improvements
|
2023-08-12 05:59:50 -06:00 |
|
Jaret Burkett
|
c7054d714f
|
Finally muted the annoying safety checker notification
|
2023-08-12 01:38:18 -06:00 |
|
Jaret Burkett
|
67dfd9ced0
|
Added inbuild plugins and made one for image referenced. WIP
|
2023-08-10 16:20:38 -06:00 |
|
Jaret Burkett
|
1a7e346b41
|
Added inbuild plugins and made one for image referenced. WIP
|
2023-08-10 16:02:44 -06:00 |
|
Jaret Burkett
|
df48f0a843
|
Moved some of the job config into base process so it will be easier to extend extensions
|
2023-08-10 12:14:05 -06:00 |
|
Jaret Burkett
|
fbc8a87a05
|
Reworked the sd rescaler script
|
2023-08-09 08:57:27 -06:00 |
|
Jaret Burkett
|
bf90740b59
|
Fixed numerous issues with traing ESRGAN
|
2023-08-08 20:03:19 -06:00 |
|
Jaret Burkett
|
ff2c9f3d04
|
Merge branch 'main' of github.com:ostris/ai-toolkit
|
2023-08-07 18:05:03 -06:00 |
|
Jaret Burkett
|
8bd536df7e
|
Added training for a custom version of ERSGAN arcitecture. Testing training now
|
2023-08-07 18:04:23 -06:00 |
|
Jaret Burkett
|
64fbd4c92a
|
Fix some windows dependency issues
|
2023-08-05 20:28:49 -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
|
7e4e660663
|
Added extensions and an example extension that merges models
|
2023-08-04 09:37:24 -06:00 |
|
Jaret Burkett
|
b865ac8b24
|
Various windows bug fixes
|
2023-08-04 05:51:58 -06:00 |
|
Jaret Burkett
|
66c6f0f6f7
|
Big refactor of SD runner and added image generator
|
2023-08-03 14:51:25 -06:00 |
|
Jaret Burkett
|
75ec5d9292
|
Hotfix to handle latest transformers clip model missing key suddenly
|
2023-08-02 12:26:18 -06:00 |
|
Jaret Burkett
|
1a25b275c8
|
Did some work on SD rescaler. Need to run a long test on it eventually.
|
2023-08-02 07:59:27 -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 |
|