40 Commits

Author	SHA1	Message	Date
Sean Sube	b931da1d2c	fix imports, lint	2023-02-26 15:21:58 -06:00
Sean Sube	584dddb5d6	lint all the new stuff	2023-02-26 14:15:30 -06:00
Sean Sube	e1d0ad54b7	lock per worker, torch before ORT	2023-02-26 12:24:51 -06:00
Sean Sube	06c74a7a96	feat(api): remove Flask app from global scope	2023-02-26 10:15:12 -06:00
Sean Sube	3626d69f40	fix(api): correctly handle missing inversion param	2023-02-25 12:54:51 -06:00
Sean Sube	2e7de16778	feat: add a way to select textual inversions	2023-02-25 11:53:32 -06:00
Sean Sube	5f3b84827b	feat(api): add batch size to txt2img and img2img pipelines (#195)	2023-02-20 08:35:18 -06:00
Sean Sube	c1189aad96	feat: add eta parameter (fixes #194)	2023-02-19 23:29:26 -06:00
Sean Sube	c459e96523	fix(api): report accurate size when face correction is run twice	2023-02-19 10:27:05 -06:00
Sean Sube	30f16035bc	fix upscale copy ctor	2023-02-18 22:36:26 -06:00
Sean Sube	7b8ced0f68	feat(api): prefer chain stage parameters over request parameters (#138)	2023-02-18 16:27:48 -06:00
Sean Sube	bfdb071c2d	chore(docs): explain model optimizations	2023-02-18 16:06:05 -06:00
Sean Sube	e0a62ccbb5	better ONNX optimization logging	2023-02-18 15:47:31 -06:00
Sean Sube	881b290116	return session options properly	2023-02-18 15:45:28 -06:00
Sean Sube	5b4c370a1b	feat(api): enable ONNX optimizations through env	2023-02-18 15:44:39 -06:00
Sean Sube	ff57527274	remove unused enum	2023-02-18 11:16:33 -06:00
Sean Sube	ffc4eb09f1	update reported image size when correction is run twice	2023-02-18 11:14:34 -06:00
Sean Sube	6dd0edc374	apply lint	2023-02-18 11:05:38 -06:00
Sean Sube	f051af359d	make default order match type	2023-02-18 11:05:23 -06:00
Sean Sube	ecf3c03f0f	feat: add parameter to run correction before upscaling (#132)	2023-02-18 11:04:08 -06:00
Sean Sube	3ca02d4875	fix(api): make size params iterable	2023-02-18 05:44:43 -06:00
Sean Sube	0ed4af18ad	fix(api): run torch gc alongside python (#156)	2023-02-16 18:11:35 -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	3a290822eb	feat(api): pass tile order to inpaint and outpaint pipelines	2023-02-11 18:00:18 -06:00
Sean Sube	7db5472531	fix(api): include face correction outscale in final image size	2023-02-06 17:59:34 -06:00
Sean Sube	de4e7b0dc9	feat: add face outscale as its own parameter (#111)	2023-02-06 17:13:37 -06:00
Sean Sube	fb376c6b62	feat(api): make LPW an image parameter	2023-02-05 17:15:37 -06:00
Sean Sube	4a728cb9ae	lint(api): include more params in JSON response and file	2023-02-05 10:50:15 -06:00
Sean Sube	54dd34d211	lint(api): apply black and isort style	2023-02-05 07:53:35 -06:00
Sean Sube	9d1f9412f6	log device options	2023-02-04 21:26:11 -06:00
Sean Sube	04a2faffd9	fix(api): set CUDA device in ORT session	2023-02-04 21:17:39 -06:00
Sean Sube	2c9a10e2b7	format devices in logs	2023-02-04 13:56:17 -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	b1eec6907a	fix(api): write border and upscale to params file	2023-02-01 22:31:01 -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	4547bcef50	feat(api): switch to python logging	2023-01-28 17:09:19 -06:00
Sean Sube	bb3a7dc0e9	log pipeline timing, add common sizes	2023-01-28 14:56:06 -06:00
Sean Sube	ba3e450d43	fix scale for outpaint	2023-01-28 09:55:47 -06:00
Sean Sube	caafc9ebc9	lint(api): make modules for chain pipeline and params	2023-01-27 22:48:06 -06:00