17 Commits

Author	SHA1	Message	Date
Jaret Burkett	adc31ec77d	Small updates and bug fixes for various things	2025-06-03 20:08:35 -06:00
Jaret Burkett	b6d25fcd10	Improvements to vae trainer. Adjust denoise prediction of DFE v3	2025-05-30 12:06:47 -06:00
Jaret Burkett	34f4c14cd6	Work on vae trainer	2025-05-28 07:42:48 -06:00
Jaret Burkett	3f3636b788	Bug fixes and little improvements here and there.	2024-06-08 06:24:20 -06:00
Jaret Burkett	b68c3ef734	Added te aug adapter	2024-02-21 21:30:26 -07: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	d3ad195b51	Added rescaling, locon, sdxl, all kinds of stuff. sdxl is still weird	2023-07-26 16:19:50 -06:00
Jaret Burkett	e6fb0229bf	Added better optimizer chooised and param support	2023-07-24 09:21:58 -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	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