1881 Commits

Author	SHA1	Message	Date
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
Sean Sube	bdefaec0d2	serialize metadata	2024-01-03 23:27:40 -06:00
Sean Sube	8ae6069ac3	handle 0 progress	2024-01-03 23:19:56 -06:00
Sean Sube	7a5e6ab895	pass result on to command	2024-01-03 23:17:35 -06:00
Sean Sube	fac98ab239	keep result of each stage with metadata	2024-01-03 23:13:21 -06:00
Sean Sube	e0d0933092	fix missing param	2024-01-03 22:20:27 -06:00
Sean Sube	46bcd5af86	make response types match	2024-01-03 22:15:50 -06:00
Sean Sube	c4b831fe5c	fix typos	2024-01-03 21:45:13 -06:00
Sean Sube	9b5e894898	avoid waiting for final progress	2024-01-03 21:39:19 -06:00
Sean Sube	28fc2082c7	track results after each stage	2024-01-03 21:31:41 -06:00
Sean Sube	4f1bc84fd9	initialize fields	2024-01-03 21:20:46 -06:00
Sean Sube	0c504e3f69	report stage and tile count in progress	2024-01-03 21:16:44 -06:00
Sean Sube	9d05d9baac	return output names	2024-01-03 20:54:11 -06:00
Sean Sube	492bc1e992	make sure image response is array	2024-01-03 20:28:53 -06:00
Sean Sube	d87861cb29	fix params, save metadata	2024-01-03 20:22:38 -06:00
Sean Sube	444d809eff	fix progress, unwrap metadata	2024-01-03 19:18:56 -06:00
Sean Sube	a6baf369d4	avoid reusing routes	2024-01-03 19:15:33 -06:00
Sean Sube	44a8d61082	feat: add batch endpoints for cancel and status, update responses	2024-01-03 19:09:18 -06:00
Sean Sube	19c91f70f5	add networks to metadata	2024-01-02 22:14:21 -06:00
Sean Sube	46098960d8	attempt to save grid mode metadata	2024-01-02 21:49:22 -06:00
Sean Sube	297285fb6f	fix(api): upscale during highres regardless of upscale checkbox	2024-01-02 21:34:11 -06:00
Sean Sube	4edd39740b	add image metadata to stage results	2024-01-02 21:24:27 -06:00
Sean Sube	0fa7eff8a8	make thumbnail size configurable	2024-01-02 21:24:09 -06:00
Sean Sube	5ad7b4c1b4	fix thumbnail call	2024-01-02 21:19:34 -06:00
Sean Sube	5de28d0361	generate a thumbnail when the first image is > 512	2024-01-02 20:58:02 -06:00
Sean Sube	35371d33fe	chore(release): 0.12.0	2023-12-31 14:15:31 -06:00
Sean Sube	00db9a2787	fix(api): make sure ControlNet constructor is used for same pipeline (#449)	2023-12-31 13:00:34 -06:00
Sean Sube	b7f2313489	add upscale param, update XL ref	2023-12-31 11:19:39 -06:00
Sean Sube	de8ab6f1c9	fix(api): use correct part of the gfpgan output	2023-12-31 07:04:43 -06:00
Sean Sube	c6718a3e3b	fix(api): make sure upscaling only runs once regardless of correction options	2023-12-31 06:39:00 -06:00
Sean Sube	71fbc871a6	feat: add API parameter for upscale checkbox	2023-12-30 19:34:40 -06:00
Sean Sube	a5ecb59cd9	fix: download LoRAs and other networks by default	2023-12-30 19:33:40 -06:00
Sean Sube	54db63394f	arch adjustments	2023-12-30 13:50:14 -06:00
Sean Sube	366efd1a58	fix RRDB rescale arch	2023-12-30 13:34:12 -06:00
Sean Sube	6834b716ea	switch RRDB nets based on upscaling	2023-12-30 13:28:16 -06:00
Sean Sube	0ddc16288f	feat(api): support both ESRGAN variants	2023-12-30 13:11:56 -06:00
Sean Sube	6cee411e43	add ESRGAN input adjustments back	2023-12-30 12:42:50 -06:00
Sean Sube	7abe6dc6a9	feat(api): enable 1x upscaling models	2023-12-30 11:59:52 -06:00
Sean Sube	11e643bcb5	remove broken channel adjustment	2023-12-30 11:59:37 -06:00
Sean Sube	477747cced	better support for ESRGAN 1x models	2023-12-30 11:50:28 -06:00
Sean Sube	698058018a	add test flag to increase time limit	2023-12-30 11:49:59 -06:00
Sean Sube	f6134dafc6	fix errors in local std denoising	2023-12-29 23:19:35 -06:00
Sean Sube	6a004816af	fix(api): make version-safe imports compatible with tests	2023-12-29 23:09:00 -06:00
Sean Sube	05ab396b2a	revert back to NL means for panorama only	2023-12-29 22:53:04 -06:00
Sean Sube	750a49a907	apply lint	2023-12-29 18:53:13 -06:00
Sean Sube	8b698ed4ab	Revert "wrap some diffusers imports for version safety" ... This reverts commit e7f7bbc9ea.	2023-12-29 18:52:54 -06:00
Sean Sube	da6beb31b2	switch to internal RRDBNet with x2 support	2023-12-29 08:50:27 -06:00
Sean Sube	1035915d36	apply lint	2023-12-29 08:19:58 -06:00
Sean Sube	ce90ffb0ee	use threshold range for local denoising	2023-12-29 08:18:37 -06:00
Sean Sube	973677967d	fix(api): turn off SDXL's very visible watermarking (#438)	2023-12-28 22:54:53 -06:00
Sean Sube	e7f7bbc9ea	wrap some diffusers imports for version safety	2023-12-28 22:49:17 -06:00
Sean Sube	f28fdda47a	feat(api): add stage for local standard deviation denoising for XL	2023-12-27 20:17:35 -06:00
Sean Sube	7d56689527	pass device to wrapper	2023-12-27 09:01:38 -06:00
Sean Sube	d17b946091	wrap ESRGAN helper	2023-12-27 08:47:06 -06:00
Sean Sube	404f24f9ad	compile pattern	2023-12-27 05:17:17 -06:00
Sean Sube	95886430a4	feat(api): support more RealESRGAN-based models	2023-12-27 05:08:15 -06:00
Sean Sube	f5e7b3b865	apply lint	2023-12-26 20:25:04 -06:00
Sean Sube	e8a30ae56b	feat(api): add optimization flag to ORT extended optimizations	2023-12-26 20:22:32 -06:00
Sean Sube	006f983269	fix(api): use standard op block list for SDXL	2023-12-26 20:22:09 -06:00
Sean Sube	cfe7a55935	type and test fixes	2023-12-26 20:21:34 -06:00
Sean Sube	6d2d5058d9	fix(api): reduce min VAE tile size to match min height/width	2023-12-26 08:07:39 -06:00
Sean Sube	eb8bd145c9	null checks	2023-12-26 08:07:05 -06:00
Sean Sube	2b65077d82	type lint	2023-12-26 08:06:16 -06:00
Sean Sube	ae0e8447f9	add message to unet type error	2023-12-24 23:10:08 -06:00
Sean Sube	e2c9389d6e	add fallback dtypes to unet patch	2023-12-24 22:57:02 -06:00
Sean Sube	ef256280b4	use right type mapping	2023-12-24 22:46:22 -06:00
Sean Sube	0f6a1a82a2	translate types to np	2023-12-24 22:40:07 -06:00
Sean Sube	39ee4cbfcd	handle XL UNets	2023-12-24 22:36:39 -06:00
Sean Sube	80a255397e	feat(api): use wrapped model's input types in UNet patch	2023-12-24 22:21:52 -06:00
Sean Sube	4eeb68cbcf	fix(api): allow numbers in archive filenames	2023-12-24 12:07:18 -06:00
Sean Sube	15b9b51ed5	fix(api): make sure new SD converter does not produce extra files	2023-12-24 10:47:46 -06:00