1951 Commits

Author	SHA1	Message	Date
Sean Sube	2d18ac1377	use preferred Torch device for prompt filters	2024-02-11 16:56:35 -06:00
Sean Sube	3fea317a43	put prompt filters behind a feature flag for now	2024-02-11 16:25:34 -06:00
Sean Sube	bc2eeb8503	feat(api): add prompt filter stage using GPT2 models	2024-02-11 16:23:10 -06:00
Sean Sube	acd9168b32	fix(api): save inpaint images regardless of full-res settings	2024-01-31 00:10:40 -06:00
Sean Sube	dd02b059ba	add missing file	2024-01-30 09:57:19 -06:00
Sean Sube	028547d90b	fix size import	2024-01-30 09:56:47 -06:00
Sean Sube	4b5ff69f98	fix circular imports	2024-01-30 09:55:35 -06:00
Sean Sube	e09b1ce6b8	feat: use CUDA 11.8 throughout	2024-01-30 09:29:28 -06:00
Sean Sube	f70de1ca79	correctly handle falsey args in with_args methods	2024-01-30 09:28:43 -06:00
Sean Sube	c2d45b03dc	fix(api): correct output names for inpaint pipelines	2024-01-29 20:56:25 -06:00
Sean Sube	8f7ef6dfce	remove references to /tmp	2024-01-28 20:50:22 -06:00
Sean Sube	7cfe619b4a	remove some duplicate code	2024-01-28 20:20:30 -06:00
Sean Sube	3c17fb8d50	fix(api): save correct image as thumbnail	2024-01-28 20:08:03 -06:00
Sean Sube	8b26c12439	fix(api): send correct image URLs	2024-01-28 20:04:47 -06:00
Sean Sube	8c8be8fc08	feat: save a thumbnail for every image	2024-01-28 19:45:37 -06:00
Sean Sube	8a71934d74	turn off thumbnails for release tests, fix unit tests	2024-01-28 16:12:24 -06:00
Sean Sube	83e5c99cf7	feat(api): read thumbnail parameter from request	2024-01-28 15:20:31 -06:00
Sean Sube	7c6ae6a094	fix(api): make with_args methods accept and ignore extra args	2024-01-28 15:15:24 -06:00
Sean Sube	f8bdd76bea	fix bad import	2024-01-28 13:12:05 -06:00
Sean Sube	964000e3a6	reduce blur when using latent mirroring	2024-01-28 13:09:33 -06:00
Sean Sube	256714b661	fix(api): add missing callback params to stages	2024-01-28 13:09:06 -06:00
Sean Sube	bdbe6549bc	fix(api): correct gradient for vertical latent mirroring	2024-01-27 21:51:36 -06:00
Sean Sube	c2ec5f9b2c	add feature flag for vertical mirroring	2024-01-27 18:05:46 -06:00
Sean Sube	d8a0f5f15f	use pipeline type to decide scheduler patch options	2024-01-27 17:48:14 -06:00
Sean Sube	ddb8f4fadf	disable latent mirroring for img2img pipelines	2024-01-27 14:18:40 -06:00
Sean Sube	2773ab0965	put latent mirroring behind feature flag, fix means	2024-01-27 09:25:00 -06:00
Sean Sube	bdf6f401a6	add gradient to mirroring	2024-01-21 22:27:34 -06:00
Sean Sube	23018a79a3	include flipped mask	2024-01-21 22:05:58 -06:00
Sean Sube	e41fa04fe9	convert mirrored latents back to torch	2024-01-21 21:53:45 -06:00
Sean Sube	285c672718	fix output type	2024-01-21 21:49:24 -06:00
Sean Sube	40dac93e18	fix gradient point order	2024-01-21 21:45:51 -06:00
Sean Sube	51217eae8a	add scheduler patch	2024-01-21 21:36:39 -06:00
Sean Sube	dcaadf1a31	add experimental patch with latent mirroring	2024-01-21 21:34:58 -06:00
Sean Sube	ff11d75784	remove timestep pinning	2024-01-15 08:01:03 -06:00
Sean Sube	4633e7ed05	fix last step errors	2024-01-14 21:59:48 -06:00
Sean Sube	bb1b3095c8	increment scheduler index after view loop	2024-01-14 21:46:47 -06:00
Sean Sube	d7af93c34a	remove panorama denoise stage	2024-01-14 21:36:54 -06:00
Sean Sube	374efca776	attempt to freeze scheduler timestep for panorama	2024-01-14 21:29:18 -06:00
Sean Sube	482d9b246d	pass correct scale params	2024-01-14 20:45:28 -06:00
Sean Sube	0f12c37931	save metadata to S3	2024-01-14 20:03:16 -06:00
Sean Sube	effa26c73b	fix(api): remove watermark calls from SDXL panorama	2024-01-14 19:51:06 -06:00
Sean Sube	4de917db3a	fix(api): include networks in launch scripts	2024-01-14 19:50:49 -06:00
Sean Sube	2faac57b93	limit status batch size	2024-01-14 19:25:19 -06:00
Sean Sube	2c988b8a16	lint fixes	2024-01-14 19:24:50 -06:00
Sean Sube	5ffb44c8fa	reset outscale for thumbnail metadata	2024-01-14 15:36:21 -06:00
Sean Sube	c47e91f71f	apply lint	2024-01-14 13:19:09 -06:00
Sean Sube	0a1ccf2d45	use adjusted metadata for thumbnail	2024-01-14 13:03:50 -06:00
Sean Sube	4b7bd2b9f7	fix thumbnail metadata	2024-01-14 12:48:24 -06:00
Sean Sube	943f0e6f04	fix thumbnail metadata size	2024-01-14 12:24:59 -06:00
Sean Sube	5e40ba949e	cache horde safety models	2024-01-14 09:59:19 -06:00
Sean Sube	8f31d16fa3	use correct upscale params for highres metadata	2024-01-14 09:49:59 -06:00
Sean Sube	f7b171f198	fix highres output size	2024-01-14 09:41:54 -06:00
Sean Sube	150ed7564d	add network tokens to experimental prompt parser	2024-01-13 20:54:59 -06:00
Sean Sube	5b5dd184f1	fix exif method	2024-01-13 20:19:37 -06:00
Sean Sube	301628e636	fix progress recursion	2024-01-13 11:32:41 -06:00
Sean Sube	a5fe52d2a2	feat(api): parse metadata from input images	2024-01-13 10:01:50 -06:00
Sean Sube	065df23902	more lint	2024-01-12 23:41:16 -06:00
Sean Sube	1cf8c7eff6	make pipeline progress more accurate and complete	2024-01-12 23:28:52 -06:00
Sean Sube	746f2f56cd	avoid mutating existing progress objects	2024-01-12 22:59:08 -06:00
Sean Sube	26c1631709	lint fixes	2024-01-12 22:58:52 -06:00
Sean Sube	c5d64e7b1e	lint fixes	2024-01-12 22:29:39 -06:00
Sean Sube	7966f001e1	fix(api): make sure a controlnet has been selected when using that pipeline	2024-01-12 22:28:59 -06:00
Sean Sube	f5506b17f0	feat(api): support CPU model offloading specifically for SDXL	2024-01-12 18:58:26 -06:00
Sean Sube	5888cae298	add settings slice and allow resizing image history	2024-01-11 21:53:02 -06:00
Sean Sube	055f6e2956	fix controlnet	2024-01-09 09:21:51 -06:00
Sean Sube	0d1439734c	fix output stage	2024-01-08 22:54:41 -06:00
Sean Sube	99bd6b0708	keep safe images	2024-01-08 22:32:49 -06:00
Sean Sube	2eb90ba559	feat: better error messaging	2024-01-08 22:14:32 -06:00
Sean Sube	0215cb9ac6	cancel unsafe jobs	2024-01-08 21:50:36 -06:00
Sean Sube	a42d728006	feat(api): add optional horde safety stage	2024-01-08 21:37:44 -06:00
Sean Sube	3101be8b78	feat: add motd to server config, show in web UI	2024-01-08 21:03:48 -06:00
Sean Sube	237accc973	start adding queue progress	2024-01-07 08:16:13 -06:00
Sean Sube	fd92d52339	send empty progress with initial job	2024-01-06 20:32:38 -06:00
Sean Sube	be40797e59	use txt2img params for highres output	2024-01-06 17:40:36 -06:00
Sean Sube	21666abf03	fix up highres metadata	2024-01-06 17:20:37 -06:00
Sean Sube	ec48b27f93	start inheriting metadata	2024-01-06 16:59:02 -06:00
Sean Sube	9e201fc94a	fix blend sources	2024-01-06 16:27:25 -06:00
Sean Sube	9b9a614f44	return results for multiple jobs	2024-01-06 15:10:07 -06:00
Sean Sube	e8e2b92436	keep metadata when tiling	2024-01-06 14:27:55 -06:00
Sean Sube	3a647ad9bd	provide empty metadata for pipelines with source images	2024-01-06 14:17:26 -06:00
Sean Sube	3e5a95548b	fix array sizing, update release tests	2024-01-06 02:33:01 -06:00
Sean Sube	1a8d538bfe	handle empty results	2024-01-05 21:23:05 -06:00
Sean Sube	058a0f1c0a	use max size of all images in stack	2024-01-05 21:19:48 -06:00
Sean Sube	b5ed1101b4	use result size helper	2024-01-05 21:16:10 -06:00
Sean Sube	e958eff3a8	misc testing fixes	2024-01-05 21:15:33 -06:00
Sean Sube	7d782842ed	fix(api): only hash networks once per worker lifetime	2024-01-05 21:11:44 -06:00
Sean Sube	6b0b2e41a6	update tests	2024-01-05 20:13:57 -06:00
Sean Sube	88f99ef6c2	remove unused format field from upscale params	2024-01-05 20:13:16 -06:00
Sean Sube	9b5d00a66a	move progress state to worker context	2024-01-05 20:12:41 -06:00
Sean Sube	4f230f4111	clean up result and metadata handling	2024-01-05 20:11:58 -06:00
Sean Sube	10acad232c	estimate steps better, pass progress onto reply	2024-01-04 19:48:43 -06:00
Sean Sube	4dc251cf4a	fix callback access	2024-01-04 19:39:44 -06:00
Sean Sube	9a1389d03b	fix imports	2024-01-04 19:16:40 -06:00
Sean Sube	b6da935be6	use progress type in command	2024-01-04 19:09:52 -06:00
Sean Sube	ce84dfa115	pass metadata through tiles	2024-01-04 09:12:32 -06:00
Sean Sube	112d5a8876	start passing metadata through stages	2024-01-04 09:01:19 -06:00
Sean Sube	e3a6635ca2	remove unused job type field	2024-01-03 23:58:57 -06:00
Sean Sube	48671e27b0	fix hash field	2024-01-03 23:46:50 -06:00
Sean Sube	fe6b267483	prep models field	2024-01-03 23:38:44 -06:00
Sean Sube	04be627e7b	pass server when serializing metadata	2024-01-03 23:32:33 -06:00