46 Commits

Author	SHA1	Message	Date
Sean Sube	8a71934d74	turn off thumbnails for release tests, fix unit tests	2024-01-28 16:12:24 -06:00
Sean Sube	964000e3a6	reduce blur when using latent mirroring	2024-01-28 13:09:33 -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	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	6a004816af	fix(api): make version-safe imports compatible with tests	2023-12-29 23:09:00 -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	1035915d36	apply lint	2023-12-29 08:19:58 -06:00
Sean Sube	e7f7bbc9ea	wrap some diffusers imports for version safety	2023-12-28 22:49:17 -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	abeeddeeb2	fix(api): handle models with integer timestamps	2023-12-23 22:19:15 -06:00
Sean Sube	9b4ae0916b	lint and test fixes	2023-12-03 12:53:50 -06:00
Sean Sube	10fab12cd0	dedupe new latent resizing code	2023-12-03 11:11:28 -06:00
Sean Sube	56f19256b5	fix(api): correctly load text encoder 2 and VAE without LoRAs	2023-09-24 09:49:16 -05:00
Sean Sube	d48dbf7d6e	fix(api): enable Unet patch for SDXL	2023-09-20 19:28:34 -05:00
Sean Sube	78f834a678	apply lint	2023-09-10 11:26:55 -05:00
Sean Sube	0fa03e77ad	feat: add panorama pipeline for SDXL	2023-09-10 11:26:18 -05:00
Sean Sube	2a160411a6	get type from ORT model	2023-06-05 23:26:11 -05:00
Sean Sube	6cb7ed58be	apply lint	2023-06-05 23:18:13 -05:00
Sean Sube	395a632946	fix(api): use VAE model dtype when converting sample	2023-06-05 23:16:34 -05:00
Sean Sube	3b02fc5768	feat: add UNet stride as its own parameter	2023-05-03 19:15:05 -05:00
Sean Sube	98386cb385	fix(api): update latent window size on VAE patches	2023-05-02 19:57:26 -05:00
Sean Sube	95725fff79	feat(api): add tile size and stride to image parameters	2023-05-01 23:21:25 -05:00
Sean Sube	47c7f5d1da	convert VAE moments back to np	2023-04-29 16:02:09 -05:00
Sean Sube	b1a989b71c	apply lint	2023-04-29 16:00:28 -05:00
Sean Sube	b4dfc44bc6	fix(api): return correct output from tiled VAE	2023-04-29 15:59:39 -05:00
Sean Sube	6d05f40ecb	correct param names for VAE encoder wrapper	2023-04-28 20:37:59 -05:00
Sean Sube	9a2421ee47	fixes for VAE encoder wrapper	2023-04-28 16:06:39 -05:00
Sean Sube	1d9de2c45b	apply lint	2023-04-28 13:56:36 -05:00
Sean Sube	79115e5f23	remove conv breaking VAE wrapper	2023-04-28 13:55:02 -05:00
Sean Sube	577e2320f5	fix sample size, use first output	2023-04-28 13:30:37 -05:00
Sean Sube	878e29ad3d	remove gradients from tiled VAE tensors	2023-04-27 23:41:43 -05:00
Sean Sube	fbeeae692b	remove config ref, add np/torch conversions	2023-04-27 23:03:17 -05:00
Sean Sube	64a753e064	feat(api): add tiled VAE wrapper	2023-04-27 22:50:11 -05:00