226 Commits

Author	SHA1	Message	Date
Sean Sube	bfdaae952f	fix(api): move optimum model validation behind a feature flag, disabled by default	2024-02-11 16:57:05 -06:00
Sean Sube	006f983269	fix(api): use standard op block list for SDXL	2023-12-26 20:22:09 -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
Sean Sube	29ccd16437	fix(api): handle HF repos in new SD converter	2023-12-24 07:05:56 -06:00
Sean Sube	da6e813638	rename old converter	2023-12-24 06:24:45 -06:00
Sean Sube	5d31e85709	use torch suffix consistently	2023-12-24 06:16:37 -06:00
Sean Sube	48630b3dfe	add torch suffix to temp dir	2023-12-24 06:12:43 -06:00
Sean Sube	525fec24c5	go back to working dir	2023-12-24 06:10:15 -06:00
Sean Sube	769350115c	pass correct paths to torch extraction	2023-12-24 06:00:44 -06:00
Sean Sube	9ffe266384	fix(api): unload UNet before CNet conversion	2023-12-24 05:32:39 -06:00
Sean Sube	054e5c406a	fully rename old converter to legacy	2023-12-24 05:31:55 -06:00
Sean Sube	9a077fae15	feat(api): replace legacy SD converter with optimum-based one by default	2023-12-23 22:55:52 -06:00
Sean Sube	a5a60d3e15	fix(api): GC during SDXL conversion	2023-12-23 22:34:05 -06:00
Sean Sube	f6e6e31789	add controlnet support to optimum converter	2023-12-23 22:33:46 -06:00
Sean Sube	ebe813d035	fix(api): change weights filename for new models to match optimum	2023-12-23 22:18:35 -06:00
Sean Sube	14208de393	register optimum-based SD converter	2023-12-23 22:18:09 -06:00
Sean Sube	b6ef00e437	feat(api): add new optimum-based SD converter	2023-12-23 22:09:57 -06:00
Sean Sube	2b8b59a39c	make SD conversion more like SDXL	2023-12-23 21:42:36 -06:00
Sean Sube	2546084208	lint(api): move some noisy log messages to trace level	2023-12-21 21:07:02 -06:00
Sean Sube	e09631ef05	apply lint	2023-12-19 23:01:05 -06:00
Sean Sube	c8a9734acf	dump unet to torch before making cnet	2023-12-19 22:50:18 -06:00
Sean Sube	4a87fb2a31	update forwarding shapes, apply lint	2023-12-19 22:10:37 -06:00
Sean Sube	b4262eb777	remove bulky controlnet wrapper	2023-12-19 20:56:24 -06:00
Sean Sube	ba9982af51	fix(api): various controlnet fixes	2023-12-19 19:19:48 -06:00
Sean Sube	133c4a20bd	fix(api): use correct post-fetch path when converting from checkpoints (#432)	2023-12-16 13:17:24 -06:00
Sean Sube	95d8f4a598	fix(api): make sure all file types are covered (#432)	2023-12-14 20:04:08 -06:00
Sean Sube	4da4cd95a5	fix(api): trim whitespace from model names because it breaks things (#376)	2023-12-10 13:59:47 -06:00
Sean Sube	9c1fcd16fa	fix(api): only fetch diffusion models if they have not already been converted (#398)	2023-12-10 13:52:52 -06:00
Sean Sube	e7da2cf8a6	fix(api): load pretrained VAE from original path	2023-12-08 23:19:52 -06:00
Sean Sube	4fd50984f0	fix(api): correct VAE extension check during conversion	2023-12-08 22:55:51 -06:00
Sean Sube	46d9fc0dd4	fix(api): make sure diffusion models have a valid prefix	2023-12-08 18:49:18 -06:00
Sean Sube	9b4ae0916b	lint and test fixes	2023-12-03 12:53:50 -06:00
Sean Sube	44a23a6366	apply lint	2023-11-24 22:40:01 -06:00
Sean Sube	d7c95a4a4f	fix SDXL node names once	2023-11-24 16:51:03 -06:00
Sean Sube	74832fc61b	fix(api): match SDXL keys per LoRA	2023-11-24 15:22:07 -06:00
Sean Sube	98f8abbacd	fix(api): correctly detect unknown embedding format	2023-11-23 23:53:17 -06:00
Sean Sube	ddd3d45b79	fix(api): add logging along SDXL custom VAE codepath	2023-11-23 23:21:45 -06:00
Sean Sube	f00bfe9bd0	more tests, apply lint	2023-11-23 11:19:58 -06:00
Sean Sube	54dc970a7e	correctly remove operator types	2023-11-21 23:12:24 -06:00
Sean Sube	33bd67beb6	switch LoRA back to fixing node names on the fly	2023-11-21 22:33:17 -06:00
Sean Sube	a39fe1d21c	use pre-fixed names for XL LoRA key matching	2023-11-21 21:46:34 -06:00
Sean Sube	60615e097b	fix(api): support normal LoRA keys in SDXL LoRAs	2023-11-21 19:10:11 -06:00
Sean Sube	7e6749e0d7	apply lint, add missing file	2023-11-18 18:13:13 -06:00
Sean Sube	eb3f1479f2	fix(api): only use optimum's fp16 mode for SDXL export when torch fp16 is enabled	2023-11-16 21:45:50 -06:00
Sean Sube	4a2498ad8d	apply lint	2023-11-08 22:04:15 -06:00
Sean Sube	f4f3bda6f8	fix(api): allow all supported tensors extensions for VAE files	2023-11-08 22:00:56 -06:00
Sean Sube	7c67d595fb	add more misc logging	2023-11-06 08:48:35 -06:00
Sean Sube	b365660adc	move more SDXL LoRA logs to trace level	2023-11-05 16:35:32 -06:00
Sean Sube	046803fc2b	lint(api): reduce log level during SDXL LoRA blending	2023-11-05 15:55:49 -06:00
Sean Sube	e9b1375440	fix(api): write tests for embedding/inversion blending	2023-10-06 19:04:48 -05:00