15 Commits

Author	SHA1	Message	Date
Jaret Burkett	bfe29e2151	Removed all submodules. Submodule free now, yay.	2025-04-18 10:39:15 -06:00
Jaret Burkett	c446f768ea	Huge memory optimizations, many big fixes	2023-08-27 17:48:02 -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	8bd536df7e	Added training for a custom version of ERSGAN arcitecture. Testing training now	2023-08-07 18:04:23 -06:00
Jaret Burkett	c35b78f0d4	Added random weight adjuster to prevent overfitting	2023-07-29 17:14:14 -06:00
Jaret Burkett	c7640b0865	WIP diffusers pipeline is weird. Starting to hate sdxl	2023-07-27 17:35:24 -06:00
Jaret Burkett	d3ad195b51	Added rescaling, locon, sdxl, all kinds of stuff. sdxl is still weird	2023-07-26 16:19:50 -06:00
Jaret Burkett	cb70c03273	SDXL should be working, but I broke something where it is not converging.	2023-07-25 13:50:59 -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	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	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	439310e4dc	Made a very basic vae trainer.	2023-07-17 19:03:50 -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