87 Commits

Author	SHA1	Message	Date
Sean Sube	82487f5771	fix(api): separate Real ESRGAN upsampler from GFPGAN stage (#87)	2023-02-12 13:50:28 -06:00
Sean Sube	034be3266e	fix(api): accumulate progress from inpaint pipelines (#90)	2023-02-12 13:16:17 -06:00
Sean Sube	2f6a3afddb	pass progress on to most stages	2023-02-12 12:33:36 -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	8ea33e9874	fix(api): patch various download fns to use cache (#95)	2023-02-12 06:25:44 -06:00
Sean Sube	ffbe0929e6	apply lint	2023-02-11 18:10:36 -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	b1a1a2b9c4	lint(api): apply style fixes	2023-02-07 08:20:02 -06:00
Sean Sube	b6fd288842	fix(api): calculate outpaint overlap factor before adding margins	2023-02-06 22:49:58 -06:00
Sean Sube	a9f4e19616	chore(api): log overlap factor when outpainting	2023-02-06 22:43:58 -06:00
Sean Sube	de4a3818a0	fix(api): use SRVGG net for Real ESRGAN v3	2023-02-06 21:36:20 -06:00
Sean Sube	ae5cf1fd28	fix(api): restore denoise code path for Real ESRGAN v3 (#109)	2023-02-06 18:11:45 -06:00
Sean Sube	651acf6991	lint recent changes	2023-02-06 17:26:51 -06:00
Sean Sube	de4e7b0dc9	feat: add face outscale as its own parameter (#111)	2023-02-06 17:13:37 -06:00
Sean Sube	564cfc1279	lint(api): remove unused upsampler arg	2023-02-06 08:56:34 -06:00
Sean Sube	24ab52bf0d	fix(api): remove background upscaler from GFPGAN stage (#108)	2023-02-06 08:33:36 -06:00
Sean Sube	1dfd9e72ef	fix(api): run background ESRGAN on same device as GFPGAN	2023-02-06 08:10:36 -06:00
Sean Sube	c7e0041229	fix(api): pass current device when loading GFPGAN	2023-02-06 08:07:06 -06:00
Sean Sube	811b6640a8	fix(api): only run one form of outpaint tiling	2023-02-05 23:02:06 -06:00
Sean Sube	d6c1244d72	fix(api): use grid tiling when inpaint margin is 0	2023-02-05 22:57:52 -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	9c84d853c9	add missing lpw param	2023-02-05 17:40:10 -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	6fe278c744	fix(api): switch between spiral and grid tiling based on outpaint margins (#101)	2023-02-05 16:56:11 -06:00
Sean Sube	ca2f0a6404	fix(api): pass outscale factor to CodeFormer (fixes #100)	2023-02-05 16:22:04 -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	7b506cb6d3	feat(api): enable LPW custom pipeline (#27)	2023-02-05 11:31:40 -06:00
Sean Sube	0a9f108156	feat(api): add CodeFormer to automatic upscale	2023-02-05 10:49:20 -06:00
Sean Sube	bf1378bf96	fix(api): log stacktrace from job errors	2023-02-05 08:50:26 -06:00
Sean Sube	e059f11253	feat(api): add CodeFormer stage for chain pipelines	2023-02-05 08:37:47 -06:00
Sean Sube	35681efc1b	clean up draft CodeFormer stage	2023-02-05 08:06:50 -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	8a81e8b810	add missing arg to chain pipeline	2023-02-04 14:27:27 -06:00
Sean Sube	ea69753bfd	fix(api): additional logging around model load ops	2023-02-02 23:34:02 -06:00
Sean Sube	aff3a3f2ee	lint(api): fix interpolation in some log messages	2023-02-02 18:52:31 -06:00
Sean Sube	a1298f1565	fix(api): respect model selection when running inpaint/outpaint stages	2023-02-02 18:51:54 -06:00
Sean Sube	6ce13096ed	feat(api): server setting to select image format	2023-02-02 08:19:57 -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	c34ddacf55	fix(api): pin outscale for GFPGAN to 1 to avoid sparse tiling	2023-01-31 17:08:30 -06:00
Sean Sube	1ca7edb4a8	fix(api): variable name in GFPGAN	2023-01-31 08:24:10 -06:00
Sean Sube	39f84c4a49	fix(api): convert images to ndarray before passing to GFPGAN	2023-01-31 08:16:57 -06:00