586 Commits

Author	SHA1	Message	Date
Sean Sube	37b173d0d1	fix(api): unwrap state dict from VAE	2023-02-17 08:23:12 -06:00
Sean Sube	c74d22aa42	fix(api): continue converting other models after an error in one (#166)	2023-02-17 07:49:45 -06:00
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	1ea1a57706	add VAE to extras schema	2023-02-16 20:55:05 -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	9e479795fa	fix(api): add sources section to extras schema	2023-02-16 19:01:26 -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	82ab8ca172	chore(release): 0.7.1	2023-02-15 21:02:22 -06:00
Sean Sube	6a2bac1359	lint(api): remove unused import	2023-02-15 21:01:25 -06:00
Sean Sube	9db684ad5d	chore(release): 0.7.0	2023-02-15 20:59:37 -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	4b1857883f	fix(api): split up base and extras launch scripts	2023-02-13 23:15:43 -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	1de591e15f	fix(api): make tmux launch script executable	2023-02-13 17:32:57 -06:00
Sean Sube	45ad18cefe	feat(api): add tmux launch script	2023-02-13 08:43:28 -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	62fec58f78	change loading steps phrasing	2023-02-12 13:32:40 -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
Sean Sube	2f6a3afddb	pass progress on to most stages	2023-02-12 12:33:36 -06:00
Sean Sube	9d1c5dca52	fix(api): correctly handle partial extras files	2023-02-12 12:23:03 -06:00
Sean Sube	06d4f32574	make progress callbacks optional	2023-02-12 12:22:11 -06:00
Sean Sube	d9fc908592	feat(api): collect progress from chain pipelines (#90)	2023-02-12 12:17:36 -06:00
Sean Sube	27a3fa8f51	feat: show model in image card, use labels for model and scheduler (#104)	2023-02-12 09:51:35 -06:00
Sean Sube	9c5043e9d0	fix(api): correctly cache diffusers scheduler	2023-02-12 09:33:13 -06:00
Sean Sube	1179092028	feat(api): intercept model downloads in libs and use cached copy (fixes #95)	2023-02-12 09:28:37 -06:00
Sean Sube	8ea33e9874	fix(api): patch various download fns to use cache (#95)	2023-02-12 06:25:44 -06:00
Sean Sube	6750200e6b	remove empty subdir readmes	2023-02-11 22:19:18 -06:00
Sean Sube	1f06b502d5	fix(api): always fill inpaint/outpaint mask with white	2023-02-11 18:55:22 -06:00
Sean Sube	ffbe0929e6	apply lint	2023-02-11 18:10:36 -06:00
Sean Sube	0356c317c9	fix(gui): get tile orders from server params	2023-02-11 18:10:02 -06:00
Sean Sube	3a290822eb	feat(api): pass tile order to inpaint and outpaint pipelines	2023-02-11 18:00:18 -06:00
Sean Sube	51651abd08	feat(gui): add tile order as an inpaint parameter (#107)	2023-02-11 17:41:58 -06:00
Sean Sube	44c9524cd2	apply lint	2023-02-11 16:50:57 -06:00
Sean Sube	6d243e8f4f	select correct platform	2023-02-11 16:26:00 -06:00
Sean Sube	66c2ce3074	fix device iteration	2023-02-11 16:19:08 -06:00
Sean Sube	db06f4a9c3	fix(api): remove any from device pool	2023-02-11 16:16:40 -06:00
Sean Sube	6b30ccc60c	log more about selected device	2023-02-11 16:14:02 -06:00
Sean Sube	38c8a8a413	make any a real device	2023-02-11 16:06:14 -06:00
Sean Sube	f45a0154e8	add any option to platforms after checking providers	2023-02-11 16:01:55 -06:00
Sean Sube	e4785b94e0	keep provider list	2023-02-11 15:55:30 -06:00
Sean Sube	9551e4a9b9	feat(api): make any platform optional	2023-02-11 15:53:27 -06:00
Sean Sube	d7383d1101	fix(api): include any platform option	2023-02-11 15:45:28 -06:00
Sean Sube	ea3b065d80	feat(api): add option to use any available platform	2023-02-11 15:41:42 -06:00
Sean Sube	917f5bedd6	lint fix	2023-02-11 15:24:12 -06:00
Sean Sube	9171ab68bc	chore(api): link to diffusers and d8ahazard licenses	2023-02-11 15:02:27 -06:00
Sean Sube	5eae80cbdd	lint(api): remove unused hub download code path	2023-02-11 14:36:17 -06:00
Sean Sube	157067b554	lint(api): fix some Sonar issues	2023-02-11 14:19:42 -06:00
Sean Sube	454abcdddc	lint(api): move half-precision CUDA check before models, apply lint	2023-02-11 13:31:34 -06:00
Sean Sube	694d15547f	lint(api): clean up conversion code from original diffusers, drop pydantic dep	2023-02-11 12:36:54 -06:00
Sean Sube	c599385a30	lint(api): apply to original diffusers converter	2023-02-10 23:32:16 -06:00
Sean Sube	2cc604823b	fix(api): launch with included extras if user does not provide their own	2023-02-10 23:06:30 -06:00
Sean Sube	9f202486c2	feat(api): add a way to download models from civitai or other https sources (#117)	2023-02-10 22:41:24 -06:00
Sean Sube	b3e4076775	fix(api): run garbage collection after job errors	2023-02-09 22:04:33 -06:00
Sean Sube	45b09168dd	fix(api): move all unet tensors to the training device (#119)	2023-02-09 22:04:01 -06:00
Sean Sube	d8d5bcd927	fix(api): make venv has been loaded in launch script	2023-02-09 22:02:27 -06:00
Sean Sube	36d164a281	add safetensors dep	2023-02-09 00:01:48 -06:00
Sean Sube	beb3411811	load local extras by default on Windows	2023-02-08 23:41:41 -06:00
Sean Sube	acb1513cb4	add temporary deps	2023-02-08 23:40:57 -06:00