433 Commits

Author	SHA1	Message	Date
Sean Sube	b3c8fce16b	fix(api): fallback to PyTorch if tensors fail to load with JIT	2023-02-17 07:39:48 -06:00
Sean Sube	005650a9a2	feat(api): log CUDA memory stats (#141)	2023-02-16 23:51:17 -06:00
Sean Sube	d42de16a84	feat(api): support custom VAE for diffusers models	2023-02-16 22:52:25 -06:00
Sean Sube	388eb640c0	apply lint, rename lookup table	2023-02-16 22:22:46 -06:00
Sean Sube	f5ae9dd492	add ORT type table	2023-02-16 21:50:19 -06:00
Sean Sube	3d73b9e621	fix(api): update SD upscaling pipeline	2023-02-16 21:44:33 -06:00
Sean Sube	873276f1d0	load checkpoint properly	2023-02-16 20:23:10 -06:00
Sean Sube	ca1b22d44d	fix(api): support loading VAE from CKPT files	2023-02-16 20:18:42 -06:00
Sean Sube	4b6be765a6	copy VAE without translation	2023-02-16 19:24:02 -06:00
Sean Sube	cddbc87ca3	skip key removal for VAE tensors	2023-02-16 19:17:00 -06:00
Sean Sube	aa6dc334d7	resolve VAE files within model path	2023-02-16 18:56:54 -06:00
Sean Sube	2c66cc58c5	feat(api): add option for custom VAE to extras file (#145)	2023-02-16 18:53:50 -06:00
Sean Sube	5e9dfa3465	lint(api): lowercase log messages	2023-02-16 18:42:05 -06:00
Sean Sube	0ed4af18ad	fix(api): run torch gc alongside python (#156)	2023-02-16 18:11:35 -06:00
Sean Sube	6a2bac1359	lint(api): remove unused import	2023-02-15 21:01:25 -06:00
Sean Sube	841b18576c	apply lint	2023-02-15 20:50:12 -06:00
Sean Sube	d60ec52186	fix(api): report correct image size for blend tab	2023-02-15 18:15:35 -06:00
Sean Sube	32a4bbcd64	fix(api): remove alpha channel after blending but before upscaling (fixes #148)	2023-02-15 17:49:42 -06:00
Sean Sube	b5eeb44c62	fix(api): remove inpaint post-multiply (#150)	2023-02-15 17:45:25 -06:00
Sean Sube	df5fb93bc0	fix(api): bypass model cache entirely when limit is 0	2023-02-15 17:17:28 -06:00
Sean Sube	2b29b099f0	fix(api): make SD upscaling compatible with more schedulers	2023-02-15 17:16:20 -06:00
Sean Sube	3e5edb1c39	fix(api): restore python 3.8 compatibility (#146)	2023-02-14 21:23:16 -06:00
Sean Sube	4d0cd2e981	apply lint	2023-02-14 19:01:14 -06:00
Sean Sube	8a2a9174ba	fix(api): pass both device and session options to ORT (#38)	2023-02-14 18:57:50 -06:00
Sean Sube	feb4603171	fix(api): remove finished jobs from worker pool (#124)	2023-02-14 17:23:23 -06:00
Sean Sube	38f8aa38ee	lint(api): read blending sources in loop	2023-02-14 17:13:29 -06:00
Sean Sube	d473a0fd2d	fix(api): pass device options to ORT session (#38)	2023-02-14 17:12:52 -06:00
Sean Sube	f2d0025545	feat: add iPNDM scheduler	2023-02-14 07:27:51 -06:00
Sean Sube	1289c560bd	apply lint	2023-02-14 00:12:07 -06:00
Sean Sube	7f6fa22b4e	fix(api): skip model download if final converted version already exists (fixes #139)	2023-02-13 23:03:51 -06:00
Sean Sube	4cc1d63cec	fix(api): make diffusion model config optional again	2023-02-13 22:49:50 -06:00
Sean Sube	6b6f2f0eff	feat: add support for DEIS multistep scheduler	2023-02-13 22:37:54 -06:00
Sean Sube	05756b2b08	fix(api): patch more download paths (#134)	2023-02-13 22:23:17 -06:00
Sean Sube	d1b250660e	fix(api): convert blend sources to the same size as the mask	2023-02-13 21:57:47 -06:00
Sean Sube	c61f200bc2	fix(api): base model cache size on number of workers	2023-02-13 18:46:33 -06:00
Sean Sube	c432ab0795	remove oldest items from model cache first	2023-02-13 18:33:06 -06:00
Sean Sube	c6b2751cc1	fix(api): patch download fn in facexlib (#134)	2023-02-13 18:30:09 -06:00
Sean Sube	6d503ca00e	update cache entries with tag and key	2023-02-13 18:27:20 -06:00
Sean Sube	51d53597c0	fix(api): load Real ESRGAN PTH from cache	2023-02-13 18:21:05 -06:00
Sean Sube	8dc1ea7f1f	declare missing cache	2023-02-13 18:15:02 -06:00
Sean Sube	f43f793148	fix type	2023-02-13 18:12:40 -06:00
Sean Sube	0709c1dbf0	feat(api): add GFPGAN and Real ESRGAN to model cache	2023-02-13 18:10:11 -06:00
Sean Sube	e9472bc005	feat(api): add model cache for diffusion models	2023-02-13 18:04:46 -06:00
Sean Sube	7fa1783be4	feat: implement blend tab and copy buttons (#62)	2023-02-13 17:34:42 -06:00
Sean Sube	d6201c9d32	feat(api): add an option for custom checkpoint config to extras file (fixes #130)	2023-02-12 14:10:30 -06:00
Sean Sube	82487f5771	fix(api): separate Real ESRGAN upsampler from GFPGAN stage (#87)	2023-02-12 13:50:28 -06:00
Sean Sube	ae8aa2ea7a	apply lint	2023-02-12 13:32:50 -06:00
Sean Sube	034be3266e	fix(api): accumulate progress from inpaint pipelines (#90)	2023-02-12 13:16:17 -06:00
Sean Sube	aaf82a42f1	fix(api): use correct var for diffusion scheduler cache	2023-02-12 12:47:55 -06:00
Sean Sube	feb7e7d57b	pass progress to txt2img and img2img	2023-02-12 12:38:06 -06:00