51 Commits

Author	SHA1	Message	Date
Sean Sube	8a2a9174ba	fix(api): pass both device and session options to ORT (#38)	2023-02-14 18:57:50 -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	6b6f2f0eff	feat: add support for DEIS multistep scheduler	2023-02-13 22:37:54 -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	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	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	3a290822eb	feat(api): pass tile order to inpaint and outpaint pipelines	2023-02-11 18:00:18 -06:00
Sean Sube	4f71348f98	feat(api): convert from SD checkpoints (#117)	2023-02-08 22:35:54 -06:00
Sean Sube	6fcfe4f878	fix(api): correct order for img2img params	2023-02-05 22:48:07 -06:00
Sean Sube	7462c96616	apply lint fixes again	2023-02-05 17:55:04 -06:00
Sean Sube	20467aafac	chore(api): add logging for LPW pipelines	2023-02-05 17:53:51 -06:00
Sean Sube	5f35a2853b	fix(api): fully switch between LPW and regular ONNX pipelines	2023-02-05 17:36:00 -06:00
Sean Sube	f3983a7917	fix(api): switch RNG based on LPW parameter	2023-02-05 17:24:15 -06:00
Sean Sube	fb376c6b62	feat(api): make LPW an image parameter	2023-02-05 17:15:37 -06:00
Sean Sube	7c5f9baeb7	fix(api): embed LPW pipeline (fixes #96)	2023-02-05 16:01:11 -06:00
Sean Sube	49b3aa68bb	fix(api): update SD upscale pipeline	2023-02-05 15:54:17 -06:00
Sean Sube	cb3590720b	lint(api): remove unused imports	2023-02-05 15:33:56 -06:00
Sean Sube	401df84069	fix(api): replace some numpy RNGs with torch equivalent	2023-02-05 13:43:33 -06:00
Sean Sube	3231a904c1	fix txt2img call	2023-02-05 11:31:44 -06:00
Sean Sube	7b506cb6d3	feat(api): enable LPW custom pipeline (#27)	2023-02-05 11:31:40 -06:00
Sean Sube	54dd34d211	lint(api): apply black and isort style	2023-02-05 07:53:35 -06:00
Sean Sube	37dd8927bf	fix(api): pass device ID in provider params	2023-02-04 21:52:45 -06:00
Sean Sube	cc12cb0fcf	clean up lingering provider refs	2023-02-04 21:23:34 -06:00
Sean Sube	04a2faffd9	fix(api): set CUDA device in ORT session	2023-02-04 21:17:39 -06:00
Sean Sube	d636ce3eef	fix(api): avoid setting device on schedulers that do not support it	2023-02-04 17:21:12 -06:00
Sean Sube	3637f642c6	fix(api): pass job context and device to upscaling	2023-02-04 14:52:23 -06:00
Sean Sube	98b6e4dd03	feat(api): add provider for each available CUDA device (#38)	2023-02-04 13:49:34 -06:00
Sean Sube	6426cff741	feat(api): switch to device pool for background workers	2023-02-04 10:06:22 -06:00
Sean Sube	ea69753bfd	fix(api): additional logging around model load ops	2023-02-02 23:34:02 -06:00
Sean Sube	a1ef6c4c77	lint(api): consolidate output fns	2023-02-02 08:31:35 -06:00
Sean Sube	0557ab9a2e	lint(api): use constant for latents scale factor	2023-02-01 22:21:22 -06:00
Sean Sube	6697c2eb6a	feat(api): save image params along with image (#81)	2023-02-01 22:20:40 -06:00
Sean Sube	066b1a0230	fix(api): run garbage collection after each model change	2023-02-01 21:20:48 -06:00
Sean Sube	6c9b0c9c1e	fix(api): remove reference to undeclared var	2023-01-31 21:29:22 -06:00
Sean Sube	a3eed7a095	fix(api): drop unused parameter from inpaint pipeline	2023-01-31 21:14:43 -06:00
Sean Sube	301cbe8b8c	pass provider to upscale pipeline, clean up some debug	2023-01-29 18:54:35 -06:00
Sean Sube	bacce0ace9	testing ONNX upscaling	2023-01-29 18:42:05 -06:00
Sean Sube	ca613cabe1	more ONNX upscaling	2023-01-29 16:57:18 -06:00
Sean Sube	8c6d957a53	continue converting upscale to ONNX	2023-01-29 15:23:19 -06:00
Sean Sube	8d346cbed0	start porting SD upscale to ONNX	2023-01-29 13:49:30 -06:00
Sean Sube	2033cbb601	fix tile latent axis	2023-01-29 11:39:02 -06:00
Sean Sube	50d6dbb451	feat(api): switch inpaint mode to upscale stage	2023-01-29 10:39:10 -06:00
Sean Sube	483a1bad68	feat(api): use per-tile latents for outpainting	2023-01-29 10:31:22 -06:00