1878 Commits

Author	SHA1	Message	Date
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
Sean Sube	29ccd16437	fix(api): handle HF repos in new SD converter	2023-12-24 07:05:56 -06:00
Sean Sube	27bd307cd8	fix(api): remove redundant torch version pin	2023-12-24 06:34:10 -06:00
Sean Sube	e707bc8dde	re-enable torch extraction by default in legacy converter	2023-12-24 06:25:24 -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	abeeddeeb2	fix(api): handle models with integer timestamps	2023-12-23 22:19:15 -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	5447d8c159	add optimum converter to schema	2023-12-23 22:10:38 -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	50d51a3921	fix(api): make argument for skipping base models work correctly	2023-12-23 14:21:39 -06:00
Sean Sube	52065ef317	lint(api): remove some confusing log messages	2023-12-23 12:17:23 -06:00
Sean Sube	972f9de2ef	bump version numbers for v0.11 release	2023-12-22 21:06:57 -06:00
Sean Sube	3d245a663d	fix(api): correct origin point when blending images (#445)	2023-12-22 11:31:31 -06:00
Sean Sube	bc9276c253	lint imports	2023-12-21 21:34:03 -06:00
Sean Sube	ccff9a3136	fix(api): silence controlnet registration warnings	2023-12-21 21:30:49 -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	7acdee89e1	fix tests	2023-12-21 20:53:16 -06:00
Sean Sube	6bec3b0cb9	fix(api): change default web UI path to match docs	2023-12-21 20:41:52 -06:00
Sean Sube	a6cb630206	fix(docs): update tile parameters in user guide	2023-12-21 20:39:01 -06:00
Sean Sube	0aab1301aa	fix(api): revert torch to 2.0.1 for ROCm due to MIopen errors (#444)	2023-12-21 18:57:00 -06:00
Sean Sube	ee3573750b	fix blend test refs	2023-12-21 09:51:26 -06:00
Sean Sube	855698cf39	fix tests	2023-12-21 09:51:06 -06:00
Sean Sube	2c2eda8c3c	fix(api): resize blend mask to match source	2023-12-20 23:41:54 -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	9a3384b348	Merge branch 'main' of github.com:ssube/onnx-web	2023-12-20 09:21:56 -06:00
Sean Sube	9c215dd90b	fix controlnet with fp16 again	2023-12-19 23:16:19 -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	a716f6db1f	fix(api): add loading path for controlnet	2023-12-18 22:21:33 -06:00
Sean Sube	6708efa8ce	make controlnet params match others	2023-12-18 22:14:33 -06:00
Sean Sube	de4b6882f1	update blend refs to RGB mode	2023-12-18 22:04:38 -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	1163accb49	fix(api): add missing param to controlnet pipeline	2023-12-18 21:39:49 -06:00
Sean Sube	a0ee620336	fix(api): update controlnet lib	2023-12-18 21:18:43 -06:00
Sean Sube	0e7e4795f5	fix(api): update arpeggio parser and boto for s3	2023-12-18 21:18:24 -06:00
Sean Sube	53fd62e8c3	pin codeformer lib	2023-12-18 21:17:57 -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
BZ Libby	fc67809612	clean-up ... I missed removing a character in the regex for the wildcard token	2023-12-18 19:09:23 -06:00
BZLibby	b807530117	fixes support for multiple wildcards in one token	2023-12-18 19:01:18 -06:00
Sean Sube	6a6a3f04bc	feat(api): update pinned onnxruntime to 0.16.3	2023-12-17 19:55:52 -06:00
Sean Sube	9e734706f1	fix(api): run blend pipeline in RGB mode (#439)	2023-12-17 14:54:10 -06:00
Sean Sube	68a3f13f62	fix(tests): add release tests for SDXL, LCM, and turbo	2023-12-17 14:17:09 -06:00
Sean Sube	76e28d962d	fix(api): parse embedding/TI tokens correctly	2023-12-17 12:48:20 -06:00
Sean Sube	41b8ae0359	open archives in read-only mode	2023-12-17 12:06:43 -06:00
Sean Sube	a65e0fd602	update schemas for archive converter	2023-12-16 22:49:29 -06:00
Sean Sube	f36fa6b5ec	bump server version in logs	2023-12-16 22:40:04 -06:00
Sean Sube	f3e1beaa71	feat(api): add converter to extract ZIP archives (#437)	2023-12-16 22:14:43 -06:00
Sean Sube	2a9265987e	start updating docs for v0.11	2023-12-16 17:03:25 -06:00
Sean Sube	b3cf0acfe9	fix(api): add debugpy to dev deps	2023-12-16 15:38:39 -06:00
Sean Sube	0364543bad	fix(api): update HF hub version (#433)	2023-12-16 15:36:32 -06:00
Sean Sube	2653dd812e	feat(api): allow embeddings as a synonym for TI token	2023-12-16 15:27:43 -06:00
Sean Sube	02dd17aa4f	fix(api): allow spaces in wildcard filenames	2023-12-16 15:21:49 -06:00
Sean Sube	cdbdd9b4e2	feat(gui): add wildcard menu to web UI	2023-12-16 15:17:28 -06:00
BZLibby	90bc28d06e	apply style	2023-12-16 14:56:19 -06:00
BZLibby	6a561173a7	adds support for dynamic wildcards in prompts	2023-12-16 14:37:56 -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	695eeaf303	feat(api): remove deprecated Karras Ve scheduler (#189)	2023-12-14 22:36:52 -06:00
Sean Sube	c6de25682d	feat(api): upgrade to latest diffusers, optimum, transformers (#433)	2023-12-14 20:13:58 -06:00
Sean Sube	6c4f4f334f	always use alpha in blend stage	2023-12-14 20:12:39 -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	2a641b111e	fix(api): check for web UI files in Windows launch scripts	2023-12-10 17:03:54 -06:00
Sean Sube	0155236744	fix(api): update flask and pin werkzeug (#414)	2023-12-10 14:10:09 -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	c9b1df9fdd	use dest in file client if provided	2023-12-10 13:30:27 -06:00
Sean Sube	e91e08484b	handle query params better in civitai client	2023-12-10 13:17:37 -06:00
Sean Sube	e200fe9186	avoid pickling environ	2023-12-10 13:07:02 -06:00
Sean Sube	9b883de1cb	pass context to client ctor	2023-12-10 12:16:01 -06:00
Sean Sube	3b32cd4ac3	sonar lint	2023-12-10 11:45:52 -06:00
Sean Sube	4edb32aaac	feat(api): add env vars for Civitai and Huggingface auth tokens	2023-12-10 11:08:02 -06:00
Sean Sube	e6978243cc	fix download location for networks	2023-12-10 10:57:25 -06:00
Sean Sube	d9c5c1bd45	lint	2023-12-09 23:51:43 -06:00
Sean Sube	2fc5ec930c	feat(api): add support for DPM SDE scheduler	2023-12-09 23:51:09 -06:00
Sean Sube	50db19922a	fix type of client instance	2023-12-09 23:40:47 -06:00
Sean Sube	419b2811ef	feat: update min CFG for SDXL turbo	2023-12-09 23:40:27 -06:00
Sean Sube	22e0597916	pass kwargs on to client	2023-12-09 23:09:00 -06:00
Sean Sube	35c973e55f	pass correct metadata to network converter	2023-12-09 23:07:56 -06:00
Sean Sube	52e78748c8	keep model field optional	2023-12-09 23:05:41 -06:00
Sean Sube	e052578a20	lint, give civitai client a name	2023-12-09 23:03:41 -06:00
Sean Sube	ebb5a586ce	get file root path from context, avoid downloading entire HF repos	2023-12-09 22:53:29 -06:00
Sean Sube	6cdba4cebb	handle extensions correctly	2023-12-09 21:05:16 -06:00
Sean Sube	8bb76f1fee	keep protocol when downloading from http sources	2023-12-09 20:50:41 -06:00
Sean Sube	4c5bb906e8	pass conversion context to generic sources	2023-12-09 20:36:04 -06:00
Sean Sube	20b719fdff	special downloading for embeds	2023-12-09 19:15:28 -06:00
Sean Sube	7496613f4e	fix fetch calls	2023-12-09 18:46:47 -06:00
Sean Sube	cc8e564d26	split up download clients, add to plugin exports	2023-12-09 18:04:34 -06:00