1
0
Fork 0
Commit Graph

403 Commits

Author SHA1 Message Date
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
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