1
0
Fork 0
Commit Graph

1663 Commits

Author SHA1 Message Date
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 d79af78ef0
fix schema 2023-11-12 15:19:04 -06:00
Sean Sube 44e483322e
feat: show tokens for networks in prompt 2023-11-12 15:15:06 -06:00
Sean Sube 3ffbc00390
fix(api): turn alternatives back off for SDXL 2023-11-12 14:23:02 -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 4513fa3428
fix edge logic 2023-11-11 21:21:12 -06:00
Sean Sube 196ef46750
correct region edges 2023-11-11 20:24:02 -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 0d1bf44d25
fix reseed size and mask dims 2023-11-10 20:00:50 -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 f4f3bda6f8
fix(api): allow all supported tensors extensions for VAE files 2023-11-08 22:00:56 -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
HoopyFreud 63dfec6770
Fixed inpaint issue with no tile_mask
Currently inpainting does not work because the tiler needs to generate a tile_mask. This solves the issue by sending any pipeline that has sources through the tiler.
2023-11-06 16:53:16 -05: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 911f87f7ec
fix(api): only consume one retry per error 2023-11-05 19:23:42 -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 b365660adc
move more SDXL LoRA logs to trace level 2023-11-05 16:35:32 -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 046803fc2b
lint(api): reduce log level during SDXL LoRA blending 2023-11-05 15:55:49 -06:00
Sean Sube 8ba9f3c0b7
basic region prompt parsing 2023-11-05 15:46:37 -06:00
Sean Sube 44851e3785
fix: increase step limits for SDXL 2023-11-05 15:37:05 -06:00
Sean Sube 5cf7a39be0
feat(api): add experimental region prompts to SDXL panorama 2023-11-05 15:36:36 -06:00
Sean Sube 783e8eab4b
add more tests 2023-11-04 20:42:11 -05: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 e9b1375440
fix(api): write tests for embedding/inversion blending 2023-10-06 19:04:48 -05:00
Sean Sube ebdfa78737
fix(api): allow PTH tensor files, add helper to check extension 2023-10-06 19:03:15 -05:00
Sean Sube 1351b2f3ff
fix(api): allow SDXL VAE in any supported tensor format, ensure new SDXL models get hash file 2023-10-06 19:01:00 -05:00
Sean Sube 047e58c916
fix(api): continue adding tests, fix bugs encountered 2023-09-28 18:45:04 -05:00
Sean Sube 898d76e4a5
fix conditional tests 2023-09-25 22:10:47 -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 0ecae65f88
fix(api): set VAE attn processor during conversion 2023-09-24 15:02:21 -05:00
Sean Sube 6e2896f7f7
fix(api): allow SDXL txt2img models in extras file and schema 2023-09-24 10:12:40 -05:00
Sean Sube a79d835193
fix(api): remove still-random seed from stage kwargs 2023-09-24 10:12:19 -05:00
Sean Sube 5ffe8b26fa
fix(api): increase max tile size for output and grid stages to max 2023-09-24 10:11:52 -05:00
Sean Sube 539140909b
fix XL text encoder loading 2023-09-24 10:04:44 -05:00
Sean Sube 5d3a7d77a5
fix(api): use Torch pipelines while loading models for conversion 2023-09-24 10:04:21 -05:00
Sean Sube c99481f484
fix(api): load replacement VAE from single file for SD v1/v2 2023-09-24 09:49:50 -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 52fdf4f48a
fix(api): test LoRA blending code 2023-09-21 18:24:08 -05:00
Sean Sube 761bfa8364
update test for SDXL Unet wrapper 2023-09-20 21:24:16 -05:00
Sean Sube d48dbf7d6e
fix(api): enable Unet patch for SDXL 2023-09-20 19:28:34 -05:00
Sean Sube 8a97fca6d0
tests for prompt slicing, latent tiling 2023-09-16 12:44:54 -05:00
Sean Sube 466a28c13b
apply lint 2023-09-15 22:06:53 -05:00
Sean Sube af6cb1f295
more lint 2023-09-15 19:22:53 -05:00
Sean Sube 963794abaa
lint, tests 2023-09-15 19:16:47 -05:00
Sean Sube d8ea00582e
fix(build): sort imports in tests 2023-09-15 08:44:02 -05:00
Sean Sube a71298ff33
remove unused schema, lint 2023-09-15 08:40:56 -05:00
Sean Sube db82c6b760
fix(build): add vscode config for API testing 2023-09-15 08:40:06 -05:00
Sean Sube 3137a465ab
adjust node names in projection test 2023-09-15 06:56:01 -05:00
Sean Sube b851c234fe
more tests 2023-09-14 19:35:48 -05:00
Sean Sube 5b659a948a
more stage tests 2023-09-13 22:56:23 -05:00
Sean Sube 4959c01ddb
start testing tile order 2023-09-13 22:40:22 -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 4d0e24a47b
fix request log keys 2023-09-13 22:06:12 -05:00
Sean Sube b4240a288d
apply lint 2023-09-13 22:04:31 -05:00
Sean Sube e034ebfc13
add a bunch of tests 2023-09-13 22:03:39 -05:00
Sean Sube a33c88e670
fix(api): make request parsing consistent between JSON and forms 2023-09-13 17:27:44 -05:00
Sean Sube 8a5e211172
lint(api): deduplicate some imports 2023-09-13 17:27:05 -05:00
Sean Sube e7aad281b0
fix(api): only remove one retry after each tile failure 2023-09-13 17:26:43 -05:00
Sean Sube 07bea84155
syntax fix 2023-09-13 08:44:57 -05:00
Sean Sube 505e408dd6
use pipeline output count 2023-09-13 08:43:31 -05:00
Sean Sube 0a5f725efa
apply lint 2023-09-12 19:17:03 -05:00
Sean Sube 7d8819ef87
add default parameters to chain pipeline 2023-09-12 19:06:13 -05:00
Sean Sube f9acf9b50f
fix(api): report chain progress using a new callback if none was provided 2023-09-12 18:19:07 -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 fd8b9bef3b
fix type check 2023-09-11 20:47:03 -05:00
Sean Sube cf2cf51b17
fix(api): resolve XL VAE within model folder 2023-09-11 20:21:43 -05:00
Sean Sube 6fb0366f52
feat(api): make tile retries configurable 2023-09-11 18:41:59 -05:00
Sean Sube cd06f9291b
add custom VAE and fp16 support to SDXL conversion 2023-09-11 18:18:38 -05:00
Sean Sube eeebdfebcb
set unlimited tile to large value, clarify variable names 2023-09-11 17:37:20 -05:00
Sean Sube 1247cb7307
super hacky multi tiling 2023-09-11 07:28:20 -05:00
Sean Sube 0047e1a12c
add size 2023-09-10 23:20:49 -05:00
Sean Sube b0df6247b2
synthesize order 2023-09-10 23:15:01 -05:00
Sean Sube feb9ea9e1a
fix order, parse right data, use output names 2023-09-10 23:06:00 -05:00
Sean Sube 51b10de265
pass initial device to chain 2023-09-10 22:46:06 -05:00
Sean Sube fb7a0996a8
wire up per stage params 2023-09-10 22:21:48 -05:00
Sean Sube dcc0063195
pass params per stage 2023-09-10 22:19:29 -05:00
Sean Sube c6302946c0
use existing output names in output stage 2023-09-10 21:52:45 -05:00
Sean Sube dc812a4e06
pass all output names to chain job 2023-09-10 21:44:00 -05:00
Sean Sube 52521e5694
handle boolean values better 2023-09-10 21:25:23 -05:00
Sean Sube d0080c476d
read chain params per stage 2023-09-10 21:23:16 -05:00
Sean Sube 441a47a885
read chain params per stage 2023-09-10 21:21:57 -05:00
Sean Sube 93fe54577c
skip controlnet in chains for now, remove empty source image 2023-09-10 21:17:09 -05:00
Sean Sube 1fb965633e
read chain pipeline from JSON, remove new endpoint 2023-09-10 20:59:13 -05:00
Sean Sube 1a732d54b6
add endpoint for multiple image generation 2023-09-10 16:35:16 -05:00
Sean Sube 956a260db6
fix import 2023-09-10 12:15:39 -05:00
Sean Sube f2d0c2f765
apply lint 2023-09-10 11:53:36 -05:00