10 Commits

Author	SHA1	Message	Date
Jaret Burkett	836fee47a6	Fixed some mismatched weights by adjusting tolerance. The mismatch ironically made the models better lol	2023-08-29 15:20:03 -06:00
Jaret Burkett	14ff51ceb4	fixed issues with converting and saving models. Cleaned keys. Improved testing for cycle load saving.	2023-08-29 12:31:19 -06:00
Jaret Burkett	714854ee86	Hude rework to move the batch to a DTO to make it far more modular to the future ui	2023-08-29 10:22:19 -06:00
Jaret Burkett	bee0b6a235	Added converters for all stable diffusion models to convert back to ldm format from diffusers.	2023-08-28 16:12:32 -06:00
Jaret Burkett	fab7c2b04a	Fixed issue with key mapping from diffusers back to ldm	2023-08-28 14:01:26 -06:00
Jaret Burkett	8105c05c12	Added bucketting capabilities to dataloader. Finally have full planned capability. noice	2023-08-26 16:36:32 -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	55a5fcc7d9	Added method to get specific keys from model	2023-08-15 14:51:04 -06:00
Jaret Burkett	c1b1e800df	Updated module names to be compatable with ney koyha unet	2023-07-31 13:16:29 -06:00
Jaret Burkett	8d8229dfc0	Handle conversions back to ldm for saving	2023-07-18 19:34:35 -06:00