1
0
Fork 0
Commit Graph

404 Commits

Author SHA1 Message Date
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 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 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 192bd08156
apply sonar lint 2023-11-25 07:50:36 -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 8d4410305e
fix(api): convert latents to numpy before using 2023-11-24 10:36:53 -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 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 b6aed0542c
fix(api): update deprecated imports from diffusers 2023-11-21 19:10:41 -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 535b685a57
feat(api): add basic plugin system 2023-11-18 17:20: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 b31227ecb3
fix(api): make sure stage params are declared for all txt2img pipelines 2023-11-15 23:14:48 -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 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 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 1ee6761340
apply lint 2023-11-12 16:38:56 -06:00
Sean Sube 44e483322e
feat: show tokens for networks in prompt 2023-11-12 15:15:06 -06:00
Sean Sube 6eb014cec8
fix(api): enable prompt alternatives for SDXL 2023-11-12 14:12:28 -06:00
Sean Sube 09f600ab54
feat(api): skip regions on last timestep 2023-11-11 22:43:41 -06:00
Sean Sube c7cff78784
fix(api): feather all edges by default 2023-11-11 20:22:08 -06:00
Sean Sube 1fae41ddcf
feat(api): add edge options to region feathering 2023-11-11 20:17:52 -06:00
Sean Sube 5fb2de85c5
feat(api): add feature flag for single-tile panorama highres 2023-11-11 17:03:01 -06:00
Sean Sube 798fa5fc6d
fix(api): allow random seed in reseed regions 2023-11-11 14:37:23 -06:00
Sean Sube 8a94cdb385
feat(api): add tokens to reseed region 2023-11-10 18:37:42 -06:00
Sean Sube 3622ac4bfb
work on non-square regions 2023-11-09 22:42:45 -06:00
Sean Sube 42c526eb25
fix region pattern, make strength and feather both decimals 2023-11-09 18:06:59 -06:00
Sean Sube 2af584eabf
add missing feather var 2023-11-08 22:05:06 -06:00
Sean Sube 4a2498ad8d
apply lint 2023-11-08 22:04:15 -06:00
Sean Sube 59515193a1
feat(api): add edge feathering to region prompts 2023-11-08 22:00:32 -06:00
Sean Sube 633e078036
unpack mask to match latents 2023-11-08 19:07:41 -06:00
Sean Sube f564bb3f65
add regions to non-XL panorama, add feathering to SDXL regions 2023-11-08 18:51:31 -06:00
Sean Sube 7c67d595fb
add more misc logging 2023-11-06 08:48:35 -06:00
Sean Sube 408e3d725b
use regional noise sample, avoid very errors from large multipliers 2023-11-05 22:48:07 -06:00
Sean Sube 997891b255
fix coords and threshold 2023-11-05 22:19:55 -06:00
Sean Sube 2de4eb92b2
replace previous latents when region multiplier passes threshold 2023-11-05 21:41:40 -06:00
Sean Sube 05f63a32b7
flip and fix h/v coords for regions 2023-11-05 19:25:13 -06:00
Sean Sube df8b4abc77
fix region regex 2023-11-05 17:46:28 -06:00
Sean Sube 1af6a75723
replace region mode with multiplier 2023-11-05 17:28:07 -06:00
Sean Sube ee2173405a
parse region coordinates 2023-11-05 16:38:43 -06:00
Sean Sube baecb38343
fix region parsing 2023-11-05 16:25:48 -06:00
Sean Sube 8498252c75
parse region groups better 2023-11-05 16:06:49 -06:00
Sean Sube c3f4c52004
fix region prompt regex 2023-11-05 15:56:03 -06:00
Sean Sube 8ba9f3c0b7
basic region prompt parsing 2023-11-05 15:46:37 -06:00
Sean Sube 5cf7a39be0
feat(api): add experimental region prompts to SDXL panorama 2023-11-05 15:36:36 -06:00
Sean Sube e8d7d9a881
feat: split up UNet and VAE tile size and overlap/stride params 2023-11-04 20:41:58 -05:00
Sean Sube 047e58c916
fix(api): continue adding tests, fix bugs encountered 2023-09-28 18:45:04 -05:00
Sean Sube 23aa00d696
start testing pipeline loading 2023-09-25 21:57:25 -05:00
Sean Sube fc02fa6be1
fix(api): blend embeddings into second tokenizer/text encoder for SDXL 2023-09-25 18:24:16 -05:00
Sean Sube e338fcd0e0
lint(api): start renaming inversions to embeddings in code 2023-09-24 18:15:58 -05:00
Sean Sube cdb09d2b44
fix variable name 2023-09-24 18:05:48 -05:00
Sean Sube e1c2ae5b1b
serialize models before loading 2023-09-24 18:04:23 -05:00
Sean Sube a3a04fd1f4
clean up text encoder loading logic, deduplicate sessions 2023-09-24 18:01:56 -05:00
Sean Sube 85b4245cef
fix text encoder loading 2023-09-24 15:02:39 -05:00
Sean Sube 539140909b
fix XL text encoder loading 2023-09-24 10:04:44 -05: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 d11b37f0b2
move text encoder 2 loading 2023-09-23 22:59:41 -05:00
Sean Sube 6b6f63564e
lint(api): start breaking down model loading 2023-09-23 20:11:05 -05:00
Sean Sube d48dbf7d6e
fix(api): enable Unet patch for SDXL 2023-09-20 19:28:34 -05:00
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 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 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 8ce09d307b
feat(api): allow a different prompt for each highres stage 2023-09-04 21:20:45 -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 2c8ba67ecb
make highres work with SDXL and use a full-size txt2img stage 2023-09-04 21:20:23 -05:00
Sean Sube 44004730ea
skip loading component models for XL 2023-09-04 21:20:01 -05:00
Sean Sube 3b6eb6c43b
apply lint 2023-09-04 21:19:52 -05:00