1563 Commits

Author	SHA1	Message	Date
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	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	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	9e734706f1	fix(api): run blend pipeline in RGB mode (#439)	2023-12-17 14:54:10 -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	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	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	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	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
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	293a1bb184	fix(api): include SD upscaling in diffusion prefixes	2023-12-08 22:26:19 -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	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	b29837d773	fix(api): convert SD models without extracting Torch intermediate (#404)	2023-12-03 15:52:36 -06:00
Sean Sube	7c3e9c22d0	improve some logs, update test refs for panorama	2023-12-03 15:34:34 -06:00
Sean Sube	92311281df	fix(api): switch pipeline ctor based on VAE presence, improve panorama logging	2023-12-03 13:03:38 -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	b398d65624	fix(api): restore additional ORT options for multi-GPU machines	2023-12-03 12:14:05 -06:00
Sean Sube	572db45bcb	lint and type fixes	2023-12-03 12:13:45 -06:00
Sean Sube	05b1ef611f	fix latent resizing	2023-12-03 11:48:52 -06:00
Sean Sube	10fab12cd0	dedupe new latent resizing code	2023-12-03 11:11:28 -06:00
Sean Sube	c42ca9ca38	handle panorama margins for img2img and inpaint	2023-12-03 06:28:36 -06:00
Sean Sube	16911b5f87	consistenly unpack views	2023-12-03 06:19:17 -06:00
Sean Sube	5035761e66	build empty values correctly	2023-12-02 22:10:59 -06:00
Sean Sube	728f58f3e7	divide latents using latent scale	2023-12-02 22:09:11 -06:00
Sean Sube	d4611e958c	complete SD panorama tiles	2023-12-02 20:40:36 -06:00
Sean Sube	4b491ec4d5	correct resize scale	2023-12-02 20:31:52 -06:00
Sean Sube	b54a57b379	fix(api): complete panorama tiles for SD pipeline	2023-12-02 20:21:31 -06:00
Sean Sube	103d1a449a	fix(api): resize latents to complete panorama blocks	2023-12-02 20:06:27 -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	d3a4cc8fd2	fix(api): use upstream SD upscaling pipeline as base for in-tree one	2023-11-26 15:23:28 -06:00
Sean Sube	fa38b474f0	fix(api): abandon pretrained loader entirely to fix SD upscaling	2023-11-26 14:13:45 -06:00
Sean Sube	75f1a2cead	pass results object to chain pipeline from endpoint	2023-11-26 10:43:08 -06:00
Sean Sube	7ca5dc12a4	fix(api): match inpaint source and mask formats	2023-11-26 10:38:12 -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	57fc183b15	Merge remote-tracking branch 'github/main' into feat/00-plugins	2023-11-25 22:27:22 -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	1818a36c11	fix(api): correctly handle completely black mask images	2023-11-25 15:25:47 -06:00
Sean Sube	d78e843af4	fix(api): consistently handle tile size across premade pipelines	2023-11-25 14:02:42 -06:00
Sean Sube	02447f5fd6	fix(api): provide some necessary SDXL config	2023-11-25 12:29:17 -06:00
Sean Sube	6ecdae44a2	fix(api): pass correct outscale to highres stages	2023-11-25 12:25:16 -06:00
Sean Sube	b1328fdfdb	fix(api): handle images with alpha channels better	2023-11-25 07:50:54 -06:00
Sean Sube	192bd08156	apply sonar lint	2023-11-25 07:50:36 -06:00
Sean Sube	17e25a3bc3	fix list parser default	2023-11-24 23:02:22 -06:00
Sean Sube	9e930a91d5	fix(api): load lists without empty items	2023-11-24 22:40:22 -06:00
Sean Sube	44a23a6366	apply lint	2023-11-24 22:40:01 -06:00
Sean Sube	3f3811e16a	fix(api): build SDXL pipeline to avoid optimum patches	2023-11-24 17:02:21 -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	8d4410305e	fix(api): convert latents to numpy before using	2023-11-24 10:36:53 -06:00
Sean Sube	98f8abbacd	fix(api): correctly detect unknown embedding format	2023-11-23 23:53:17 -06:00
Sean Sube	34493e7968	update inpaint pipeline to use stage results	2023-11-23 23:22:28 -06:00
Sean Sube	a90a963765	fix(api): ensure pipeline is using selected scheduler	2023-11-23 23:22:05 -06:00
Sean Sube	ddd3d45b79	fix(api): add logging along SDXL custom VAE codepath	2023-11-23 23:21:45 -06:00
Sean Sube	51f0806c4a	restore inpainting fix	2023-11-23 11:36:03 -06:00
Sean Sube	3ffb30f121	fix(api): update deprecated import from diffusers	2023-11-23 11:34:18 -06:00
Sean Sube	f00bfe9bd0	more tests, apply lint	2023-11-23 11:19:58 -06:00
Sean Sube	66dfa7206a	add LCM to available scheduler list	2023-11-21 23:23:46 -06:00
Sean Sube	fd97eebfa2	feat(api): add LCM scheduler	2023-11-21 23:14:15 -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	b6aed0542c	fix(api): update deprecated imports from diffusers	2023-11-21 19:10:41 -06:00
Sean Sube	60615e097b	fix(api): support normal LoRA keys in SDXL LoRAs	2023-11-21 19:10:11 -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	781b34151c	feat(api): add flag to launch debugger	2023-11-19 21:05:39 -06:00
Sean Sube	c1f1bf2b36	various stack fixes	2023-11-19 21:05:21 -06:00
Sean Sube	8a3174aaff	load plugins from env	2023-11-19 18:41:55 -06:00
Sean Sube	98fcc07524	rewrite tile handling for image stacks	2023-11-19 18:39:39 -06:00
Sean Sube	e64e514e7f	fix(api): add out channels to upscaling config	2023-11-18 21:35:57 -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	5a517704ea	fix(tests): expand worker tests	2023-11-18 17:20:45 -06:00
Sean Sube	535b685a57	feat(api): add basic plugin system	2023-11-18 17:20:13 -06:00
Sean Sube	d52c68d607	feat(api): add chain pipeline stage result type	2023-11-18 17:19:35 -06:00
Sean Sube	c8dd85e798	fix(api): correct file extension validity check	2023-11-16 22:04:11 -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	b31227ecb3	fix(api): make sure stage params are declared for all txt2img pipelines	2023-11-15 23:14:48 -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	1bdfb4ac8a	fix(api): do not parse tokens when removing them	2023-11-14 20:34:56 -06:00
Sean Sube	14ddfb43bb	fix(api): handle empty prompts when parsing	2023-11-14 20:32:23 -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	def8ad73c5	feat(api): add feature flags, move panoramic tile feature into flags	2023-11-12 22:53:43 -06:00
Sean Sube	2a27c3ffd1	fix(api): adjust region strength threshold to support up to 90% UNet overlap	2023-11-12 21:41:19 -06:00
Sean Sube	768f478884	feat(api): add strength param to denoise stage	2023-11-12 21:30:01 -06:00
Sean Sube	21d1240e28	only add denoise stage after panorama pipeline	2023-11-12 21:19:07 -06:00
Sean Sube	95e2d6d710	feat(api): add denoise stage, use before highres	2023-11-12 21:13:52 -06:00
Sean Sube	4460625309	try replacing NaN and clipping large values to reduce noise	2023-11-12 18:24:30 -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	a4bf4ac651	add tokens to network response	2023-11-12 15:36:51 -06:00
Sean Sube	44e483322e	feat: show tokens for networks in prompt	2023-11-12 15:15:06 -06:00