14 Commits

Author	SHA1	Message	Date
Jaret Burkett	88acc28d7f	Prep for runpod docker	2024-08-03 12:41:06 -06:00
Jaret Burkett	0bc4d555c7	A lot of pixart sigma training tweaks	2024-07-28 11:23:18 -06:00
Jaret Burkett	5da3613e0b	Bug fixes and minor features	2024-04-25 06:14:31 -06:00
Jaret Burkett	34eb563d55	Added dataset tagging and management tools using llava	2023-10-24 10:11:47 -06:00
Jaret Burkett	76c764af49	Fixed issues with adapter and disabled telementry for now	2023-09-24 12:51:29 -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	ae70200d3c	Bug fixes, speed improvements, compatability adjustments withdiffusers updates	2023-09-13 07:03:53 -06:00
Jaret Burkett	d74dd636ee	Memory optimizations. Default to using cudamalloc when torch 2.0 for mem allocation	2023-09-12 04:30:23 -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	7726911562	Allow name to be passed through command line	2023-07-31 11:56:57 -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	57f14e5ef2	WIP implementing training	2023-07-12 08:23:46 -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