356 Commits

Author	SHA1	Message	Date
Sean Sube	fd28c095fd	update black, apply latest lint	2024-03-02 12:16:04 -06:00
Sean Sube	bfdaae952f	fix(api): move optimum model validation behind a feature flag, disabled by default	2024-02-11 16:57:05 -06:00
Sean Sube	6b0b2e41a6	update tests	2024-01-05 20:13:57 -06:00
Sean Sube	54db63394f	arch adjustments	2023-12-30 13:50:14 -06:00
Sean Sube	6834b716ea	switch RRDB nets based on upscaling	2023-12-30 13:28:16 -06:00
Sean Sube	0ddc16288f	feat(api): support both ESRGAN variants	2023-12-30 13:11:56 -06:00
Sean Sube	477747cced	better support for ESRGAN 1x models	2023-12-30 11:50:28 -06:00
Sean Sube	da6beb31b2	switch to internal RRDBNet with x2 support	2023-12-29 08:50:27 -06:00
Sean Sube	d17b946091	wrap ESRGAN helper	2023-12-27 08:47:06 -06:00
Sean Sube	404f24f9ad	compile pattern	2023-12-27 05:17:17 -06:00
Sean Sube	95886430a4	feat(api): support more RealESRGAN-based models	2023-12-27 05:08:15 -06:00
Sean Sube	006f983269	fix(api): use standard op block list for SDXL	2023-12-26 20:22:09 -06:00
Sean Sube	eb8bd145c9	null checks	2023-12-26 08:07:05 -06:00
Sean Sube	4eeb68cbcf	fix(api): allow numbers in archive filenames	2023-12-24 12:07:18 -06:00
Sean Sube	15b9b51ed5	fix(api): make sure new SD converter does not produce extra files	2023-12-24 10:47:46 -06:00
Sean Sube	29ccd16437	fix(api): handle HF repos in new SD converter	2023-12-24 07:05:56 -06:00
Sean Sube	e707bc8dde	re-enable torch extraction by default in legacy converter	2023-12-24 06:25:24 -06:00
Sean Sube	da6e813638	rename old converter	2023-12-24 06:24:45 -06:00
Sean Sube	5d31e85709	use torch suffix consistently	2023-12-24 06:16:37 -06:00
Sean Sube	48630b3dfe	add torch suffix to temp dir	2023-12-24 06:12:43 -06:00
Sean Sube	525fec24c5	go back to working dir	2023-12-24 06:10:15 -06:00
Sean Sube	769350115c	pass correct paths to torch extraction	2023-12-24 06:00:44 -06:00
Sean Sube	9ffe266384	fix(api): unload UNet before CNet conversion	2023-12-24 05:32:39 -06:00
Sean Sube	054e5c406a	fully rename old converter to legacy	2023-12-24 05:31:55 -06:00
Sean Sube	9a077fae15	feat(api): replace legacy SD converter with optimum-based one by default	2023-12-23 22:55:52 -06:00
Sean Sube	a5a60d3e15	fix(api): GC during SDXL conversion	2023-12-23 22:34:05 -06:00
Sean Sube	f6e6e31789	add controlnet support to optimum converter	2023-12-23 22:33:46 -06:00
Sean Sube	ebe813d035	fix(api): change weights filename for new models to match optimum	2023-12-23 22:18:35 -06:00
Sean Sube	14208de393	register optimum-based SD converter	2023-12-23 22:18:09 -06:00
Sean Sube	b6ef00e437	feat(api): add new optimum-based SD converter	2023-12-23 22:09:57 -06:00
Sean Sube	2b8b59a39c	make SD conversion more like SDXL	2023-12-23 21:42:36 -06:00
Sean Sube	50d51a3921	fix(api): make argument for skipping base models work correctly	2023-12-23 14:21:39 -06:00
Sean Sube	52065ef317	lint(api): remove some confusing log messages	2023-12-23 12:17:23 -06:00
Sean Sube	ccff9a3136	fix(api): silence controlnet registration warnings	2023-12-21 21:30:49 -06:00
Sean Sube	2546084208	lint(api): move some noisy log messages to trace level	2023-12-21 21:07:02 -06:00
Sean Sube	e09631ef05	apply lint	2023-12-19 23:01:05 -06:00
Sean Sube	c8a9734acf	dump unet to torch before making cnet	2023-12-19 22:50:18 -06:00
Sean Sube	4a87fb2a31	update forwarding shapes, apply lint	2023-12-19 22:10:37 -06:00
Sean Sube	b4262eb777	remove bulky controlnet wrapper	2023-12-19 20:56:24 -06:00
Sean Sube	ba9982af51	fix(api): various controlnet fixes	2023-12-19 19:19:48 -06:00
Sean Sube	41b8ae0359	open archives in read-only mode	2023-12-17 12:06:43 -06:00
Sean Sube	a65e0fd602	update schemas for archive converter	2023-12-16 22:49:29 -06:00
Sean Sube	f3e1beaa71	feat(api): add converter to extract ZIP archives (#437)	2023-12-16 22:14:43 -06:00
Sean Sube	133c4a20bd	fix(api): use correct post-fetch path when converting from checkpoints (#432)	2023-12-16 13:17:24 -06:00
Sean Sube	95d8f4a598	fix(api): make sure all file types are covered (#432)	2023-12-14 20:04:08 -06:00
Sean Sube	4da4cd95a5	fix(api): trim whitespace from model names because it breaks things (#376)	2023-12-10 13:59:47 -06:00
Sean Sube	9c1fcd16fa	fix(api): only fetch diffusion models if they have not already been converted (#398)	2023-12-10 13:52:52 -06:00
Sean Sube	c9b1df9fdd	use dest in file client if provided	2023-12-10 13:30:27 -06:00
Sean Sube	e91e08484b	handle query params better in civitai client	2023-12-10 13:17:37 -06:00
Sean Sube	9b883de1cb	pass context to client ctor	2023-12-10 12:16:01 -06:00