1
0
Fork 0
Commit Graph

154 Commits

Author SHA1 Message Date
Sean Sube af6cb1f295
more lint 2023-09-15 19:22:53 -05:00
Sean Sube bbff1913ef
fix(api): do not apply prompt expansion patch to SDXL 2023-09-13 22:09:41 -05:00
Sean Sube e034ebfc13
add a bunch of tests 2023-09-13 22:03:39 -05:00
Sean Sube 0a5f725efa
apply lint 2023-09-12 19:17:03 -05:00
Sean Sube 55ddb9fdac
feat(api): attempt to calculate total steps for chain pipelines 2023-09-12 18:16:16 -05:00
Sean Sube 4ccdedba89
set model path for VAE sessions 2023-09-12 07:32:29 -05:00
Sean Sube d4b013068d
fix(api): ensure VAE is loaded on correct device 2023-09-12 07:21:35 -05:00
Sean Sube 0fa03e77ad
feat: add panorama pipeline for SDXL 2023-09-10 11:26:18 -05:00
Sean Sube 761ee05134
apply lint 2023-09-05 17:36:17 -05:00
Sean Sube edfd059b46
remove SDXL inpaint import 2023-09-05 17:05:27 -05:00
Sean Sube bb9cda58ad
apply lint 2023-09-04 21:21:06 -05:00
Sean Sube 2047f7d8cf
pass XL components as ORT sessions 2023-09-04 21:21:00 -05:00
Sean Sube 9999a1bf56
add support for XL inpainting pipeline 2023-09-04 21:20:57 -05:00
Sean Sube 6b31075616
fix(api): override pipeline models if pipeline ignored components 2023-09-04 21:20:42 -05:00
Sean Sube ea9023c2eb
fix(api): only run SDXL LoRA node matching on XL models 2023-09-04 21:20:39 -05:00
Sean Sube ecac0d5770
apply lint 2023-09-04 21:20:33 -05:00
Sean Sube bbacfd1ca0
feat(api): support for SDXL LoRAs 2023-09-04 21:20:30 -05:00
Sean Sube 44004730ea
skip loading component models for XL 2023-09-04 21:20:01 -05:00
Sean Sube b1a554e1a7
fix import 2023-09-04 21:19:24 -05:00
Sean Sube a201a74b40
add SDXL pipelines to list 2023-09-04 21:19:20 -05:00
Sean Sube 47b10945ff
fix(api): use consistent cache key for each model type 2023-07-03 11:33:56 -05:00
Sean Sube dcea248b7b
fix(api): skip prompt parsing patch when using LPW 2023-06-11 07:52:46 -05:00
Sean Sube eb100f647d
fix(api): remove device and patch calls to fix SD upscaling 2023-05-09 18:22:29 -05:00
Sean Sube 98f99b1523
fix(api): update VAE tiling params every time pipeline is used 2023-05-05 20:27:27 -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 55a1e46fc3
fix(api): update panorama params when using cached pipeline 2023-05-02 18:46:49 -05:00
Sean Sube a8869c06a4
reduce tile by latent scale before passing to pipeline 2023-05-01 23:40:17 -05:00
Sean Sube 8b57767f4a
fix chain stage loading 2023-05-01 23:25:55 -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 980cba1d29
wrap VAE encoder 2023-04-28 15:56:30 -05:00
Sean Sube 1d9de2c45b
apply lint 2023-04-28 13:56:36 -05:00
Sean Sube 64a753e064
feat(api): add tiled VAE wrapper 2023-04-27 22:50:11 -05:00
Sean Sube b8b73d8592
fix(api): make sure UNet wrapper always has embeds defined 2023-04-27 09:05:38 -05:00
Sean Sube 214af7fe6a
feat(api): add panorama pipeline for ONNX 2023-04-26 07:35:20 -05:00
Sean Sube 3ce508d08f
apply lint 2023-04-24 21:04:35 -05:00
Sean Sube dc6bf330c1
feat(api): add optimization for max attention slicing (#355) 2023-04-24 19:13:32 -05:00
Sean Sube 3f4dd70b53
apply lint 2023-04-24 18:10:12 -05:00
Sean Sube cad87b97f4
feat(api): allow offloading individual models to CPU (#330) 2023-04-24 17:40:12 -05:00
Sean Sube 477fb69e03
apply lint 2023-04-23 21:16:46 -05:00
Sean Sube 0d60406eca
fix(api): make SD upscaling work with negative prompt 2023-04-23 18:22:59 -05:00
Sean Sube 8736a13277
lint(exe): move Windows bundle entry script to correct folder 2023-04-23 18:22:50 -05:00
Sean Sube 8146ef881d
only apply prompt embeds when LPW is not selected, load upscale pipeline like others 2023-04-23 18:22:41 -05:00
Sean Sube bd3e6100a2
collect prompt sections between alt groups 2023-04-23 18:22:29 -05:00
Sean Sube b249936963
feat(api): parse alternative groups in prompts 2023-04-23 15:03:11 -05:00
Sean Sube 6c00b2d87d
initialize dtype for SD wrappers 2023-04-22 12:00:09 -05:00
Sean Sube d21346e5da
apply lint 2023-04-22 10:42:22 -05:00
Sean Sube 9e9feb2ea0
lint(api): consistently use server dtype 2023-04-22 00:28:29 -05:00
Sean Sube b076326ab3
fix(api): restore inpaint pipeline option 2023-04-15 15:43:31 -05:00
Sean Sube 95841ffe2b
fix(api): check diffusers version before imports (#336) 2023-04-15 14:32:22 -05:00
Sean Sube 16c6bdf85a
use correct ORT session options 2023-04-15 12:31:28 -05:00
Sean Sube 30968f7c33
apply LoRAs to correct UNet 2023-04-15 12:28:55 -05:00
Sean Sube ac9b3d3b08
implement noise and blur source filters, move LPW in with other pipelines 2023-04-15 11:58:19 -05:00
Sean Sube 735f9d174f
apply lint 2023-04-15 11:20:11 -05:00
Sean Sube c91916b4ca
hide img2img from pipelines list 2023-04-15 11:11:43 -05:00
Sean Sube 7f16b2f0ae
clean up CNet-only conversion code 2023-04-15 09:08:14 -05:00
Sean Sube 4fa9af7922
hide unused pipeline options 2023-04-15 09:02:17 -05:00
Sean Sube c2f8ce5814
load UNet with ControlNet weights as needed 2023-04-13 23:05:00 -05:00
Sean Sube bd992398ae
lint, fix numpy error in pix2pix 2023-04-13 18:41:22 -05:00
Sean Sube 52d639ffed
use keyword args for networks 2023-04-13 00:02:47 -05:00
Sean Sube c91bdb7679
fix serialization and scheduler 2023-04-12 23:26:16 -05:00
Sean Sube 4aabf1ee27
wire up pipeline parameter, apply lint 2023-04-12 23:11:53 -05:00
Sean Sube 2af1530a7e
feat: make pipeline type a request parameter 2023-04-12 22:58:48 -05:00
Sean Sube b2556809e9
apply lint 2023-04-12 20:06:13 -05:00
Sean Sube 555ec5a644
turn controlnet into a select list and localize its label, fix name lookup on server 2023-04-12 20:03:00 -05:00
Sean Sube 948bec6d0f
load controlnets from network models list 2023-04-12 08:51:16 -05:00
Sean Sube 9e017ee35d
feat: add parameter for ControlNet selection 2023-04-12 08:43:15 -05:00
Sean Sube f49a4ddfdf
test controlnet pipe 2023-04-11 23:06:32 -05:00
Sean Sube 2c47904057
lint(api): use constant for model filename 2023-03-24 08:14:19 -05:00
Sean Sube fa71d87e2c
apply lint 2023-03-21 22:19:50 -05:00
Sean Sube e445d2afaa
fix hidden state conversion 2023-03-21 22:08:17 -05:00
Sean Sube c8aad8554e
fix(api): convert back to model format after blending, convert samples as needed (#274) 2023-03-21 22:05:14 -05:00
Sean Sube 0732058aa8
feat(api): detect Textual Inversion type from keys (#262) 2023-03-19 20:16:52 -05:00
Sean Sube b797b3d616
fix(api): do not assume file extension for additional networks 2023-03-19 15:27:51 -05:00
Sean Sube 1c631c28d3
feat(api): add flag for ORT float16 optimizations 2023-03-19 11:59:35 -05:00
Sean Sube edd32f6044
apply lint, fix up log levels 2023-03-19 10:25:09 -05:00
Sean Sube 6f283c5c02
proxy nets that need fp16 conversion 2023-03-19 10:18:22 -05:00
Sean Sube bbd4c0fd72
dtype logging for potential fp16 params 2023-03-19 09:59:08 -05:00
Sean Sube e5862d178c
fix(api): assume inversion tokens are embeddings for now 2023-03-18 18:35:11 -05:00
Sean Sube af326a784f
fix(api): avoid loading encoder twice when using LoRAs and inversions together 2023-03-18 13:39:35 -05:00
Sean Sube 9f9b73b780
fix(api): use ORT session for correct device when loading blended nets 2023-03-18 13:39:04 -05:00
Sean Sube 1f6105a8fe
make blend functions take tuples rather than split lists 2023-03-18 10:50:48 -05:00
Sean Sube 8e8e230ffd
apply lint 2023-03-15 19:27:29 -05:00
Sean Sube 506cf9f65f
feat(api): blend Textual Inversions from prompt 2023-03-15 17:14:52 -05:00
Sean Sube 973ad0f682
fix inversion var 2023-03-15 08:57:17 -05:00
Sean Sube b451196dfa
add prefix to inversion path 2023-03-15 08:55:35 -05:00
Sean Sube 421ed632b1
only use name from first inversion 2023-03-15 08:53:15 -05:00
Sean Sube a2e21f427f
feat(api): load Textual Inversions from prompt 2023-03-15 08:51:12 -05:00
Sean Sube 829cedc934
fix error when prompt has no LoRA tokens 2023-03-15 08:35:44 -05:00
Sean Sube a7f77a033d
feat(api): parse LoRA weights from prompt 2023-03-15 08:30:31 -05:00
Sean Sube 45166f281e
apply lint 2023-03-14 23:32:47 -05:00
Sean Sube 91210ee236
remove extra param, correct output path 2023-03-14 23:32:18 -05:00
Sean Sube 143904fc51
feat(api): parse LoRA names from prompt 2023-03-14 22:28:18 -05:00
Sean Sube 03f4e1b922
start wiring LoRAs into prompt 2023-03-14 22:10:33 -05:00
Sean Sube ce05e76947
load base models from model dir, use correct ORT provider 2023-03-14 21:57:37 -05:00
Sean Sube 56a4519818
test loading UNet and encoder with LoRAs 2023-03-14 21:27:23 -05:00
Sean Sube cb2b054fde
fix(api): only run GC when a diffusion model has been replaced 2023-03-11 08:06:03 -06:00
Sean Sube fe437e8cd4
feat: add support for UniPC multistep scheduler (#226) 2023-03-09 08:00:59 -06:00
Sean Sube 6e71775782
fix(api): load LPW pipeline from local source (#224) 2023-03-08 19:47:22 -06:00
Sean Sube 66c42485cb
feat(api): add support for extremely long prompts 2023-03-07 19:00:25 -06:00