451 Commits

Author	SHA1	Message	Date
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	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	bdefaec0d2	serialize metadata	2024-01-03 23:27:40 -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	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	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	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	7abe6dc6a9	feat(api): enable 1x upscaling models	2023-12-30 11:59:52 -06:00
Sean Sube	f6134dafc6	fix errors in local std denoising	2023-12-29 23:19:35 -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	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	cfe7a55935	type and test fixes	2023-12-26 20:21:34 -06:00
Sean Sube	2b65077d82	type lint	2023-12-26 08:06:16 -06:00
Sean Sube	3d245a663d	fix(api): correct origin point when blending images (#445)	2023-12-22 11:31:31 -06:00
Sean Sube	1fbee0ae52	fix(api): tile stage masks	2023-12-20 23:33:13 -06:00
Sean Sube	f6ff53d0d1	pass correct crop	2023-12-20 23:02:58 -06:00
Sean Sube	de28ecdfad	fix(api): handle blending images of different sizes better	2023-12-20 22:58:57 -06:00
Sean Sube	6e5d10f521	apply lint	2023-12-18 22:04:24 -06:00
Sean Sube	2263093c4d	convert images to cv2	2023-12-18 21:53:20 -06:00
Sean Sube	3e1db707ac	fix(api): update codeformer patches for new lib	2023-12-18 21:12:08 -06:00
Sean Sube	7ed30ee470	feat(api): switch to codeformer lib that works with torch 2.x	2023-12-18 21:06:39 -06:00
Sean Sube	6c4f4f334f	always use alpha in blend stage	2023-12-14 20:12:39 -06:00
Sean Sube	d95cfc582a	apply lint	2023-12-04 18:45:28 -06:00
Sean Sube	95a62b17ed	fix(api): detect all mask keys, immediately bubble up cancellation errors	2023-12-04 18:44:58 -06:00
Sean Sube	ae34e466ef	more lint, avoid some unnecessary errors	2023-12-03 12:57:56 -06:00
Sean Sube	9b4ae0916b	lint and test fixes	2023-12-03 12:53:50 -06:00
Sean Sube	572db45bcb	lint and type fixes	2023-12-03 12:13:45 -06:00
Sean Sube	10fab12cd0	dedupe new latent resizing code	2023-12-03 11:11:28 -06:00
Sean Sube	48d0d3e22d	use presence of stage mask to trigger tiling	2023-11-29 21:52:32 -06:00
Sean Sube	042181b9c5	fix(api): correctly handle image stacks in persist stages	2023-11-27 21:45:32 -06:00
Sean Sube	828c9511a6	fix(api): restore single-tile code path	2023-11-27 18:53:39 -06:00
Sean Sube	a9b4303d25	switch upscaling to np random	2023-11-26 15:29:07 -06:00
Sean Sube	93e3125e28	fix(api): be more careful with VAE patch flags, add margin to latents if needed	2023-11-25 23:18:57 -06:00
Sean Sube	83ebd77c3d	fix(api): handle blending numpy arrays	2023-11-25 21:19:11 -06:00
Sean Sube	1c3b2f8dfc	fix(api): better handling of alpha channels	2023-11-25 18:52:47 -06:00
Sean Sube	6ecdae44a2	fix(api): pass correct outscale to highres stages	2023-11-25 12:25:16 -06:00
Sean Sube	192bd08156	apply sonar lint	2023-11-25 07:50:36 -06:00
Sean Sube	51f0806c4a	restore inpainting fix	2023-11-23 11:36:03 -06:00
Sean Sube	c4aa262ed7	try a different hack	2023-11-19 23:40:34 -06:00
Sean Sube	65912c5a4a	apply lint to tests, test highres	2023-11-19 23:18:57 -06:00
Sean Sube	4691e80744	keep tiles after slicing	2023-11-19 21:22:55 -06:00
Sean Sube	c1f1bf2b36	various stack fixes	2023-11-19 21:05:21 -06:00
Sean Sube	98fcc07524	rewrite tile handling for image stacks	2023-11-19 18:39:39 -06:00
Sean Sube	eb77c83d80	move stages and tests to using stage result	2023-11-18 21:35:00 -06:00
Sean Sube	7e6749e0d7	apply lint, add missing file	2023-11-18 18:13:13 -06:00
Sean Sube	a63669c76b	start replacing image output with results	2023-11-18 18:08:38 -06:00
Sean Sube	d52c68d607	feat(api): add chain pipeline stage result type	2023-11-18 17:19:35 -06:00
Sean Sube	59685770e4	feat(api): disable prompt alternatives for panorama to fix regions	2023-11-14 23:23:23 -06:00
Sean Sube	5824149cc0	fix(api): use full-size latents for all panorama pipelines	2023-11-14 20:59:51 -06:00
Sean Sube	a3519f5610	reduce default denoise strength	2023-11-12 23:29:00 -06:00
Sean Sube	43944b7001	push blending out by 1	2023-11-12 23:28:47 -06:00
Sean Sube	7106e96e77	fix(api): make tile blending symmetrical, correct bottom/right gradients	2023-11-12 23:01:53 -06:00
Sean Sube	768f478884	feat(api): add strength param to denoise stage	2023-11-12 21:30:01 -06:00
Sean Sube	95e2d6d710	feat(api): add denoise stage, use before highres	2023-11-12 21:13:52 -06:00
Sean Sube	e653560f03	Merge branch 'feat/00-prompt-tokens'	2023-11-12 16:40:53 -06:00
Sean Sube	bdda39f924	Merge pull request #411 from HoopyFreud/main ... Fixed inpaint issue with no tile_mask	2023-11-12 16:40:26 -06:00
Sean Sube	1ee6761340	apply lint	2023-11-12 16:38:56 -06:00
Sean Sube	3ffbc00390	fix(api): turn alternatives back off for SDXL	2023-11-12 14:23:02 -06:00
Sean Sube	6eb014cec8	fix(api): enable prompt alternatives for SDXL	2023-11-12 14:12:28 -06:00
Sean Sube	4513fa3428	fix edge logic	2023-11-11 21:21:12 -06:00
Sean Sube	196ef46750	correct region edges	2023-11-11 20:24:02 -06:00
Sean Sube	1fae41ddcf	feat(api): add edge options to region feathering	2023-11-11 20:17:52 -06:00
Sean Sube	5fb2de85c5	feat(api): add feature flag for single-tile panorama highres	2023-11-11 17:03:01 -06:00
Sean Sube	798fa5fc6d	fix(api): allow random seed in reseed regions	2023-11-11 14:37:23 -06:00
Sean Sube	0d1bf44d25	fix reseed size and mask dims	2023-11-10 20:00:50 -06:00
Sean Sube	8a94cdb385	feat(api): add tokens to reseed region	2023-11-10 18:37:42 -06:00
Sean Sube	3622ac4bfb	work on non-square regions	2023-11-09 22:42:45 -06:00
Sean Sube	4a2498ad8d	apply lint	2023-11-08 22:04:15 -06:00
Sean Sube	59515193a1	feat(api): add edge feathering to region prompts	2023-11-08 22:00:32 -06:00
Sean Sube	633e078036	unpack mask to match latents	2023-11-08 19:07:41 -06:00
Sean Sube	f564bb3f65	add regions to non-XL panorama, add feathering to SDXL regions	2023-11-08 18:51:31 -06:00
HoopyFreud	63dfec6770	Fixed inpaint issue with no tile_mask ... Currently inpainting does not work because the tiler needs to generate a tile_mask. This solves the issue by sending any pipeline that has sources through the tiler.	2023-11-06 16:53:16 -05:00
Sean Sube	7c67d595fb	add more misc logging	2023-11-06 08:48:35 -06:00
Sean Sube	2de4eb92b2	replace previous latents when region multiplier passes threshold	2023-11-05 21:41:40 -06:00
Sean Sube	911f87f7ec	fix(api): only consume one retry per error	2023-11-05 19:23:42 -06:00
Sean Sube	e8d7d9a881	feat: split up UNet and VAE tile size and overlap/stride params	2023-11-04 20:41:58 -05:00
Sean Sube	e338fcd0e0	lint(api): start renaming inversions to embeddings in code	2023-09-24 18:15:58 -05:00
Sean Sube	5ffe8b26fa	fix(api): increase max tile size for output and grid stages to max	2023-09-24 10:11:52 -05:00
Sean Sube	963794abaa	lint, tests	2023-09-15 19:16:47 -05:00
Sean Sube	a71298ff33	remove unused schema, lint	2023-09-15 08:40:56 -05:00
Sean Sube	b851c234fe	more tests	2023-09-14 19:35:48 -05:00
Sean Sube	e034ebfc13	add a bunch of tests	2023-09-13 22:03:39 -05:00