7 Commits

Author	SHA1	Message	Date
Jaret Burkett	2e84b3d5b1	Update VAE trainer to handle fixed latent target. Also minor bug fixes and improvements	2025-07-12 16:55:15 -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	c698837241	Fixes to esrgan trainer. Moved logic for sd prompt embeddings out of diffusers pipeline so I can manipulate it	2023-09-16 17:41:07 -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	e6fb0229bf	Added better optimizer chooised and param support	2023-07-24 09:21:58 -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