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
Sean Sube
440e47f736
fix(api): defer txt2img tiling to panorama pipeline when selected
2023-09-04 21:19:45 -05:00
Sean Sube
d94a8d6f6e
fix(api): avoid padding latents when they are not being tiled
2023-09-04 21:19:42 -05:00
Sean Sube
944c92b824
fix(api): ensure panorama never generates a negative number of views
2023-09-04 21:19:39 -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
01c7392a61
add helper to remove NaN values in UNet output
2023-07-16 12:03:14 -05:00
Sean Sube
8d6ea63330
apply lint
2023-07-16 10:55:12 -05:00
Sean Sube
cac4c138b2
remove unused param
2023-07-16 10:46:26 -05:00
Sean Sube
b31f546516
lint(api): use more accurate worker name throughout
2023-07-15 18:54:54 -05:00
Sean Sube
eebbd8e7c7
apply lint
2023-07-13 07:35:20 -05:00
BZLibby
5ecc874c70
fix wildcard scripts to work on windows
2023-07-13 05:19:13 -07:00
Sean Sube
cf95ad2f0e
fix(api): match size of existing latent patch
2023-07-12 22:02:05 -05:00
Sean Sube
3d4c77d5d0
fix(api): generate new latents for partial tiles
2023-07-12 22:00:11 -05:00
HoopyFreud
6fee7f418d
Merge branch 'main' into main
2023-07-12 21:55:44 -04:00
HoopyFreud
7d9f93c7ea
Update run.py
...
Fixed full_res_inpaint check when combined with border zero check
2023-07-12 20:42:04 -04:00
Sean Sube
865b25e6d7
feat(api): support wildcards in nested folders
2023-07-11 22:29:48 -05:00
HoopyFreud
1ecc4969de
linting
2023-07-10 23:24:36 -04:00
HoopyFreud
bf7a1089d3
forgot to save
2023-07-10 23:24:05 -04:00
HoopyFreud
3c1fa6094e
full-res inptaining as parameters
2023-07-10 23:16:17 -04:00
HoopyFreud
58b5d18608
better mask shifting
2023-07-10 22:22:08 -04:00
HoopyFreud
9b630f97ee
Merge branch 'ssube:main' into main
2023-07-10 20:02:32 -04:00
HoopyFreud
b9f1e7acc1
added shifting for near-border full-res inpaint
2023-07-10 10:24:39 -04:00
HoopyFreud
2b4f640711
wrong check. don't code while sleepy.
2023-07-10 02:33:13 -04:00
HoopyFreud
e37a7fe92f
syntax error
2023-07-10 02:32:02 -04:00
HoopyFreud
b1347ed1cc
added border integrity check
2023-07-10 02:15:46 -04:00
HoopyFreud
b9603faccd
full-res inpaint
2023-07-10 02:02:30 -04:00
Sean Sube
da6f00d2a5
prevent latent tiles from going out of bounds
2023-07-09 22:35:15 -05:00
Sean Sube
60aa8ab4c0
fix(api): generate latents before tiling
2023-07-09 22:19:40 -05:00
HoopyFreud
306f7e9989
linting
2023-07-09 01:02:27 -04:00
HoopyFreud
08172a7236
new tiled inpainting method
2023-07-09 00:56:20 -04:00
HoopyFreud
757fb0d6a8
Update run.py
...
Added variable overlap to txt2img
2023-07-08 21:26:45 -04:00
HoopyFreud
098873b89c
Merge branch 'main' into main
2023-07-08 10:18:19 -04:00
HoopyFreud
cecee8653b
Update run.py
...
Use overlap in img2img
2023-07-06 22:46:03 -04:00
Sean Sube
f7fc442ba5
fix(api): keep network tokens while replacing wildcards in the saved prompt
2023-07-06 21:16:51 -05:00
Sean Sube
b8c0bb03e8
apply lint
2023-07-06 18:46:21 -05:00
Sean Sube
e65de8115e
feat(api): do not repeat wildcard values
2023-07-06 18:10:08 -05:00
Sean Sube
fed4315095
fix(api): remove unintentional tuple in img2img pipe
2023-07-05 06:50:33 -05:00
Sean Sube
0b5f554cf4
re-parse input prompt before saving metadata
2023-07-04 20:13:45 -05:00
Sean Sube
42e6f356a9
fix(api): unwrap results of img2img chain
2023-07-04 18:48:07 -05:00
Sean Sube
ac55fbd7d0
replace wildcards properly
2023-07-04 17:12:10 -05:00
Sean Sube
fa3a36ed25
fix(api): replace unknown wildcards with empty string
2023-07-04 16:54:17 -05:00
Sean Sube
c8a9dd4475
feat(api): add support for wildcards
2023-07-04 16:41:54 -05:00
Sean Sube
99a073aed2
update API entrypoints for multi-image
2023-07-04 13:56:02 -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
99c91a301c
feat: make enabling highres a parameter of its own
2023-07-02 19:08:40 -05:00
Sean Sube
d8ec93a619
feat(api): pass tile size param to most pipeline stages
2023-07-02 18:54:10 -05:00
Sean Sube
a7be651032
fix(api): remove nested tiling in highres
2023-07-02 12:39:59 -05:00
Sean Sube
4b84188f6e
fix(api): make panorama switch mode based on kwargs
2023-07-01 15:04:12 -05:00
Sean Sube
2913cd0382
feat(api): make chain stages into classes with max tile size and step count estimate
2023-07-01 12:33:14 -05:00
Sean Sube
934dabb39e
fix(api): pass mask and sources to blending pipeline in correct order
2023-07-01 12:32:39 -05:00
Sean Sube
5556478669
fix(api): pass upscale params to correction stages
2023-07-01 12:32:30 -05:00
Sean Sube
d9f251c88b
fix(api): repair progress for API pipelines
2023-07-01 12:32:05 -05:00
Sean Sube
2d10252564
fix(api): use kwargs for chain stages
2023-07-01 12:31:59 -05:00
Sean Sube
7a73c9ff61
fix(api): pass additional params to new stages
2023-07-01 12:31:41 -05:00
Sean Sube
7a951065e4
# This is a combination of 3 commits.
...
# This is the 1st commit message:
pass size to txt2img stage
# This is the commit message #2 :
pass highres and upscale params
# This is the commit message #3 :
pass correct strength param
2023-07-01 12:30:53 -05:00
Sean Sube
af416c252d
feat(api): make chain pipeline work without a source image
2023-07-01 12:30:43 -05:00
Sean Sube
fd3e65eafc
feat(api): start using chain pipelines for all images
2023-07-01 12:30:30 -05:00
Sean Sube
01811abf5a
add loras to exif params
2023-06-26 07:48:39 -05:00
Sean Sube
f142f83ad5
dump json parameters to str before writing to exif
2023-06-26 07:14:32 -05:00
Sean Sube
7e21b9539e
feat(api): embed image parameters in EXIF data ( #383 )
2023-06-26 07:03:06 -05:00
Sean Sube
12e489b761
fix(api): use correct tile size for highres upscaling
2023-06-16 21:22:00 -05:00
Sean Sube
f8d59ab65a
fix(api): swap dimensions for non-upscaling highres modes
2023-06-15 22:39:16 -05:00
Sean Sube
7b0bed7aea
fix(api): use tile size for all highres scaling methods
2023-06-14 22:52: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
7ec2597afb
apply lint
2023-06-07 07:42:56 -05:00
Sean Sube
b92853d438
fix(api): limit latent padding to tile size
2023-06-06 08:14:09 -05:00
Sean Sube
6b1ce37592
fix(api): pad partial latents when working with tiles
2023-06-05 23:44:42 -05:00
Sean Sube
2a160411a6
get type from ORT model
2023-06-05 23:26:11 -05:00
Sean Sube
6cb7ed58be
apply lint
2023-06-05 23:18:13 -05:00
Sean Sube
395a632946
fix(api): use VAE model dtype when converting sample
2023-06-05 23:16:34 -05:00
Sean Sube
ec8f367316
use overlap param, apply to spiral tiling as well
2023-06-03 20:35:21 -05:00
Sean Sube
dde1e2daad
fix(api): correctly pass noise level to SD upscaling
2023-05-09 20:25:00 -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
746e33b1f5
slice mask for views
2023-05-01 19:40:19 -05:00
Sean Sube
e8d6ab64c1
add view iteration to panoramic inpaint
2023-05-01 19:09:52 -05:00
Sean Sube
71b95182a1
feat(api): add inpaint to panorama pipeline
2023-05-01 18:53:46 -05:00
Sean Sube
5dfb09bd3c
fix(api): use existing progress counter for highres and loopback
2023-04-30 17:27:51 -05:00
Sean Sube
0573721d12
fix(api): always use accumulating progress counters
2023-04-29 23:28:51 -05:00
Sean Sube
2690eafe09
fix(api): pass correct text model type when converting v2 checkpoints ( #360 )
2023-04-29 22:45:48 -05:00
Sean Sube
4ea372d53a
fix(api): rely on pipeline cache for highres and loopback
2023-04-29 20:12:46 -05:00
Sean Sube
93b02261b2
fix(api): use img2img pipeline for loopback when controlnet was originally selected
2023-04-29 18:50:26 -05:00
Sean Sube
47c7f5d1da
convert VAE moments back to np
2023-04-29 16:02:09 -05:00
Sean Sube
b1a989b71c
apply lint
2023-04-29 16:00:28 -05:00
Sean Sube
b4dfc44bc6
fix(api): return correct output from tiled VAE
2023-04-29 15:59:39 -05:00
Sean Sube
d699c75283
fix(api): allow more LoRA/TI filenames
2023-04-29 09:09:25 -05:00
Sean Sube
8e0e6cfb9e
apply lint
2023-04-28 23:07:03 -05:00
Sean Sube
d0d154e411
fix timestamp range for panoramic img2img
2023-04-28 23:04:18 -05:00
Sean Sube
763b4f105a
apply lint
2023-04-28 20:38:09 -05:00
Sean Sube
6d05f40ecb
correct param names for VAE encoder wrapper
2023-04-28 20:37:59 -05:00
Sean Sube
9a2421ee47
fixes for VAE encoder wrapper
2023-04-28 16:06:39 -05:00
Sean Sube
980cba1d29
wrap VAE encoder
2023-04-28 15:56:30 -05:00
Sean Sube
c8e1519419
fix(api): match panorama img2img parameter check to numpy code path
2023-04-28 14:22:47 -05:00
Sean Sube
1d9de2c45b
apply lint
2023-04-28 13:56:36 -05:00
Sean Sube
79115e5f23
remove conv breaking VAE wrapper
2023-04-28 13:55:02 -05:00
Sean Sube
577e2320f5
fix sample size, use first output
2023-04-28 13:30:37 -05:00
Sean Sube
878e29ad3d
remove gradients from tiled VAE tensors
2023-04-27 23:41:43 -05:00
Sean Sube
fbeeae692b
remove config ref, add np/torch conversions
2023-04-27 23:03:17 -05:00
Sean Sube
64a753e064
feat(api): add tiled VAE wrapper
2023-04-27 22:50:11 -05:00
Sean Sube
eef85aabb8
fix param order for numpy code path
2023-04-27 21:02:43 -05:00
Sean Sube
c6fc860484
fix(api): make panorama work with prompt alternatives
2023-04-27 13:22:07 -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
bc946bb085
fix(api): use correct latents for scheduler noise
2023-04-27 09:00:45 -05:00
Sean Sube
bad00dc508
fix refs to latents
2023-04-27 08:58:53 -05:00
Sean Sube
d2baa323ab
fix(api): match LPW parameter order
2023-04-27 08:19:18 -05:00
Sean Sube
47d80b07b3
feat(api): add img2img mode to panorama pipeline
2023-04-27 07:22:00 -05:00
Sean Sube
29c616d99e
mostly working panorama
2023-04-26 21:10:13 -05:00
Sean Sube
a792bd2c5e
numpy fixes for panorama
2023-04-26 08:33:03 -05:00
Sean Sube
214af7fe6a
feat(api): add panorama pipeline for ONNX
2023-04-26 07:35:20 -05:00
Sean Sube
42a47200ef
Revert "fix(api): use padding token for prompt padding"
...
This reverts commit 2fe6a87818
.
2023-04-24 21:50:36 -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
9c337df7df
fix(api): make SD upscaling compatible with multi-prompt patch
2023-04-24 18:31:33 -05:00
Sean Sube
2fe6a87818
fix(api): use padding token for prompt padding
2023-04-24 18:10:46 -05:00
Sean Sube
3f4dd70b53
apply lint
2023-04-24 18:10:12 -05:00
Sean Sube
d0e4b8da29
feat(api): add system notifications when image jobs finish ( #329 )
2023-04-24 17:40:50 -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
b1aafd24aa
handle lack of negative prompt
2023-04-23 19:47:03 -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
4f9d12cefa
remove LoRAs and TIs from SD upscaling
2023-04-23 18:22:44 -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
0eeb342576
apply alternatives to upscale stage
2023-04-23 18:22:38 -05:00
Sean Sube
f1957f7378
fix prompt segment order
2023-04-23 18:22:36 -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
4ac25f0650
fix(api): remove traces of highres from blend pipeline
2023-04-22 12:40:51 -05:00
Sean Sube
6c00b2d87d
initialize dtype for SD wrappers
2023-04-22 12:00:09 -05:00
Sean Sube
0a75f961dc
fix(api): share loaded pipeline when running loopback or highres after img2img
2023-04-22 11:47:01 -05:00
Sean Sube
8068659c0f
handle image filter none better
2023-04-22 11:38:05 -05:00
Sean Sube
348421dd6d
fix controlnet segmentation string, add some source filter logging
2023-04-22 11:26:21 -05:00
Sean Sube
106b377a57
pass strength to loopback
2023-04-22 11:01:47 -05:00
Sean Sube
954d8e9c1c
move loopback into the right code path
2023-04-22 10:54:39 -05:00
Sean Sube
d21346e5da
apply lint
2023-04-22 10:42:22 -05:00
Sean Sube
00fb64ba82
feat: add img2img loopback ( #331 )
2023-04-22 10:39:23 -05:00
Sean Sube
7b0095a665
feat(api): add support for negative embeds ( #348 )
2023-04-22 10:05:58 -05:00
Sean Sube
9e9feb2ea0
lint(api): consistently use server dtype
2023-04-22 00:28:29 -05:00
Sean Sube
6e7f202b8d
fix(api): store both pre-parse and parsed prompts ( #320 )
2023-04-22 00:11:33 -05:00
Sean Sube
31965356d8
fix(api): handle mixed tensors for ControlNet on CPU
2023-04-15 17:03:04 -05:00
Sean Sube
dd771663ef
fix(api): add missing arguments to prompt patch ( #336 )
2023-04-15 15:56:02 -05:00
Sean Sube
b076326ab3
fix(api): restore inpaint pipeline option
2023-04-15 15:43:31 -05:00
Sean Sube
b20df68131
fix(api): switch between available pipelines
2023-04-15 15:37:45 -05:00
Sean Sube
317029356e
apply lint
2023-04-15 14:42:14 -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
6854cbf23c
add attribution for ControlNet and Pix2Pix pipelines
2023-04-15 11:44:20 -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
27954f3e65
feat(gui): add highres control to most tabs
2023-04-14 20:29:44 -05:00
Sean Sube
ad35c41c9d
feat(api): add highres to img2img mode for all pipelines
2023-04-14 08:54:21 -05:00
Sean Sube
c2f8ce5814
load UNet with ControlNet weights as needed
2023-04-13 23:05:00 -05:00
Sean Sube
06d55b0f1f
add filtered source to outputs
2023-04-13 22:51:59 -05:00
Sean Sube
428fb06ecf
fix imports, add missing controlnet deps
2023-04-13 21:10:00 -05:00
Sean Sube
80d00e4477
feat(api): add source image filters for controlnet and others
2023-04-13 20:06:33 -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
982dc10e64
fix(api): make pix2pix work with scheduler with float sigma
2023-04-12 23:44:08 -05:00
Sean Sube
eb428e9541
fix strength
2023-04-12 23:33:01 -05:00
Sean Sube
8bf888959f
fix controlnet strength param
2023-04-12 23:30:59 -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
113f21fd34
remove safety checker
2023-04-11 19:40:08 -05:00