apextoaster
/
onnx-web
1
0
Fork 0
Code Actions
2,312 Commits 135 Branches 17 Tags 132 MiB
Commit Graph

1740 Commits

Author SHA1 Message Date
Sean Sube 0ad250251e
feat(api): add an experimental multi-prompt syntax 2023-09-04 21:20:27 -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 ee430a962b
fixes for optimum 2023-09-04 21:20:16 -05:00
Sean Sube 404a314050
lint(api): add typed errors from cancelled job and download error 2023-09-04 21:20:04 -05:00
Sean Sube 44004730ea
skip loading component models for XL 2023-09-04 21:20:01 -05:00
Sean Sube 273ace7693
add helper method to get max size 2023-09-04 21:19:57 -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 8e1f188d8f
fix(api): pass replacement VAE to diffusers conversion 2023-09-04 21:19:36 -05:00
Sean Sube 58795de00d
fix(api): update minimum image size for SDXL base 2023-09-04 21:19:34 -05:00
Sean Sube f73dce507c
fix(models): update attn parameter name in CNet 2023-09-04 21:19:31 -05:00
Sean Sube 0fafc71ec8
add XL pipelines to list of valid options 2023-09-04 21:19:28 -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 c81dcf16e6
chore(release): 0.10.0 2023-08-22 20:42:10 -05:00
Sean Sube dc9048c76f
fix(api): use correct case for PNG metadata 2023-08-02 17:43:17 -05:00
Sean Sube 079d4aa602
feat(gui): load and parse most parameters from image, text, or json 2023-07-19 22:42:49 -05:00
Sean Sube 10fa36c1c1
fix(build): use 3.8 packages in AMD linux requirements file to match ROCm base image 2023-07-17 22:46:32 -05:00
Sean Sube c506fe1e9f
make sure job is always defined, remove unused placeholders from logs 2023-07-17 22:46:02 -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 9f9ef8878d
apply lint 2023-07-15 19:01:06 -05:00
Sean Sube c52ee7da4d
lint(api): replace deprecated warn method when logging 2023-07-15 19:00:52 -05:00
Sean Sube dfe9f6ee44
fix(api): split up worker name and job name to avoid logging false failures 2023-07-15 19:00:20 -05:00
Sean Sube 7f099c02e5
reset retries with each new job 2023-07-15 18:55:38 -05:00
Sean Sube b31f546516
lint(api): use more accurate worker name throughout 2023-07-15 18:54:54 -05:00
Sean Sube ada482c183
add and use retry exception type 2023-07-15 17:05:27 -05:00
Sean Sube 1942c45789
apply lint 2023-07-15 16:55:52 -05:00
Sean Sube c7e1fce557
clear model cache on stage errors 2023-07-15 12:33:07 -05:00
Sean Sube 822dfeab01
only remove retries when an error occurs 2023-07-15 12:26:37 -05:00
Sean Sube 4f0c6d292d
exit from chain pipeline when retries are exhausted 2023-07-15 12:18:25 -05:00
Sean Sube 7761aea28b
share retries throughout pipeline 2023-07-15 11:20:25 -05:00
Sean Sube d418cb1bc2
run GC during stage retries 2023-07-15 11:00:06 -05:00
Sean Sube caa96ed645
apply lint 2023-07-15 09:26:39 -05:00
Sean Sube 0a701c8d6c
add retry to stage loop 2023-07-15 09:12:37 -05:00
Sean Sube 170e1c7a70
fix(build): copy extras file into image from models dir 2023-07-14 22:50:55 -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
HoopyFreud ae6166aaf9 Update setup-nvidia.ps1 2023-07-13 05:18:54 -07:00
HoopyFreud 5578b9b66b Update setup-cpu.ps1 2023-07-13 05:18:54 -07:00
HoopyFreud 1e565c9bca Add ps1 setup script support 2023-07-13 05:18:54 -07:00
HoopyFreud 81ac14b87d Delete extras.json 
Delete extraneous extras.json
 2023-07-13 05:18:54 -07:00
HoopyFreud 74dd2b82d7 Delete launch-extras.sh 
Deleting redundant launch scripts
 2023-07-13 05:18:54 -07:00
HoopyFreud f21375f9b0 Delete launch-extras.bat 
Deleting redundant launch scripts
 2023-07-13 05:18:54 -07:00
HoopyFreud 04c7bed4eb Add PS1 (powershell) launch support 2023-07-13 05:18:54 -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
Sean Sube 4b8358b0c9
Merge remote-tracking branch 'hoopyfreud/main' 2023-07-12 21:59:39 -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 5df9aa2af7
feat(api): load wildcards from markup files 2023-07-12 18:56:48 -05: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
Sean Sube 1a084eba1c
apply lint 2023-07-10 17:48:26 -05:00
Sean Sube d9dd1e4b57
fix(api): improve handling of non-square images around tile size 2023-07-10 17:41:08 -05: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 95cad909fc
use tile size when completing partial latents 2023-07-09 23:49:34 -05:00
Sean Sube 8acfaaa1ee
ensure tiled latents are correct shape 2023-07-09 22:45:51 -05:00
Sean Sube da6f00d2a5
prevent latent tiles from going out of bounds 2023-07-09 22:35:15 -05:00
Sean Sube 76bed9582b
fixes for txt2img 2023-07-09 22:28:23 -05:00
Sean Sube 60aa8ab4c0
fix(api): generate latents before tiling 2023-07-09 22:19:40 -05:00
HoopyFreud c15f750821
Merge pull request #1 from HoopyFreud/main 
Update to latest code
 2023-07-09 14:59:58 -04:00
HoopyFreud e1d3821792
Update base.py 
Updated logging for stage run
 2023-07-09 14:31:11 -04:00
HoopyFreud 75a22cc51b
Merge branch 'main' into main 2023-07-09 14:13:54 -04:00
HoopyFreud 925c740ddf
Merge branch 'main' into tiled-inpaint 2023-07-09 14:12:51 -04:00
HoopyFreud ed03222234
Update tile.py 
Fixed extra noise source imports
 2023-07-09 14:11:25 -04:00
Sean Sube 2cc4feefe9
fix(api): save prompt with wildcards replaced in EXIF data 2023-07-09 11:51:14 -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 3fbf9baae6 linting 2023-07-08 21:30:54 -04:00
HoopyFreud 80e124554d
Update tile.py 
Accommodated no-source tiles better
 2023-07-08 21:27:28 -04:00
HoopyFreud 757fb0d6a8
Update run.py 
Added variable overlap to txt2img
 2023-07-08 21:26:45 -04:00
HoopyFreud 2d24ed8ff8 tiling in txt2img, subtiling in txt2img and img2img 2023-07-08 20:48:40 -04:00
HoopyFreud 098873b89c
Merge branch 'main' into main 2023-07-08 10:18:19 -04:00
HoopyFreud 8085015d2a linted 2023-07-08 10:17:35 -04:00
HoopyFreud 332946d8d5
Update tile.py 
Fixed logging
 2023-07-08 10:03:06 -04:00
HoopyFreud 05e8addb06
Update tile.py 
Update spiral tiling
 2023-07-06 22:46:36 -04:00
HoopyFreud cecee8653b
Update run.py 
Use overlap in img2img
 2023-07-06 22:46:03 -04:00
HoopyFreud 36b94bac82
Update params.py 
Change default tiling
 2023-07-06 22:44:53 -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 2121c7aa5d
feat(scripts): add json and parquet support to prompt book parser 2023-07-06 08:11:43 -05:00
Sean Sube 818d222ff8
apply lint 2023-07-05 23:05:47 -05:00
Sean Sube de8105ec60
feat(scripts): add script to parse prompt books 2023-07-05 23:05:00 -05:00
Sean Sube 1133321179
chore(scripts): reduce timeout for slow release tests 2023-07-05 23:04:38 -05:00
Sean Sube 17a710197c
feat(api): ignore comments and empty lines in wildcard files 2023-07-05 23:04:11 -05:00
Sean Sube 15c26cd74b
fix(api): remove tile size limit for img2img and outpaint stages 2023-07-05 17:16:30 -05:00
Sean Sube 3798ea77e7
fix(api): collect final images from SD upscale stage 2023-07-05 08:22:29 -05:00
Sean Sube fed4315095
fix(api): remove unintentional tuple in img2img pipe 2023-07-05 06:50:33 -05:00
Sean Sube c0dd4e96dc
fix(api): handle panoramic outpainting output 2023-07-04 20:44:48 -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 10050263bf
create batch size latents 2023-07-04 17:27:10 -05:00
Sean Sube 89dbf81335
fix(api): pass batch size and eta params to txt2img stage 2023-07-04 17:24:37 -05:00
Sean Sube 17b90ef2d7
use correct pipeline output 2023-07-04 17:14:25 -05:00
Sean Sube ac55fbd7d0
replace wildcards properly 2023-07-04 17:12:10 -05:00
Sean Sube a3f6fca989
init wildcards as dict 2023-07-04 16:57:44 -05:00
Sean Sube 933ee50375
load wildcards from models subdir 2023-07-04 16:56:39 -05:00
Sean Sube ec4662690f
use extension when loading wildcards 2023-07-04 16:55:26 -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 b816307994
fix(api): send tile filters as correct type 2023-07-04 16:40:02 -05:00
Sean Sube 6425a91fe3
fix(api): support multiple outputs from txt2img stage 2023-07-04 16:24:47 -05:00
Sean Sube 7c1b85f793
fix(api): define tile size for logging 2023-07-04 16:22:24 -05:00
Sean Sube 99a073aed2
update API entrypoints for multi-image 2023-07-04 13:56:02 -05:00
Sean Sube e1fcbb9093
start updating chain logic for multiple outputs 2023-07-04 13:47:31 -05:00
Sean Sube 37185252a5
feat(api): make pipeline stages support multiple images 2023-07-04 13:29:58 -05:00
Sean Sube f718087a54
lint(api): make chain pipeline logs more consistent 2023-07-04 12:27:17 -05:00
Sean Sube d1ecef9f2e
apply lint 2023-07-04 12:09:46 -05:00
Sean Sube 6ec7777f77
lint(api): type fixes and hints throughout 2023-07-04 10:20:28 -05:00
Sean Sube 5d13629ee8
fix(api): log number of tiles contributing to each image pixel 2023-07-03 11:39:48 -05:00
Sean Sube ed953f518a
feat(api): apply overlap to highres upscaling stage 2023-07-03 11:34:17 -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 a9fa76737e
fix(api): pass strength to LPW pipe from img2img blend 2023-07-03 08:27:37 -05:00
Sean Sube 3653723742
lint(api): move tile gradient logs to debug level 2023-07-03 08:05:57 -05:00
Sean Sube 197dc48b06
fix(api): make img2img blend work with LPW 2023-07-03 08:05:30 -05:00
Sean Sube 12698d38eb
fix(api): remove inner loops in upscale stages 2023-07-02 20:38:52 -05:00
Sean Sube 99c91a301c
feat: make enabling highres a parameter of its own 2023-07-02 19:08:40 -05:00
Sean Sube 3edf5e6c4d
fix(api): handle unlimited size stages correctly 2023-07-02 18:57:11 -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 c515d25dd4
fix(api): use upscaling method param in highres 2023-07-02 18:45:22 -05:00
Sean Sube 8781f7fda0
fix(api): make tiling compatible with source stages 2023-07-02 18:31:47 -05:00
Sean Sube 2a9e4a24a3
lint(api): make all other stages inherit from base stage 2023-07-02 18:21:21 -05:00
Sean Sube c9a1ace40b
fix(api): tile stages based on input image or size param 2023-07-02 18:14:52 -05:00
Sean Sube d98ecc023d
fix(api): do not overlap upscale tiles during highres 2023-07-02 17:13:16 -05:00
Sean Sube 5d18779e27
fix(api): add progress bars to release tests 2023-07-02 13:18:00 -05:00
Sean Sube a7be651032
fix(api): remove nested tiling in highres 2023-07-02 12:39:59 -05:00
Sean Sube eef055eddd
fix(api): log tile size and overlap 2023-07-02 05:15:01 -05:00
Sean Sube 4b84188f6e
fix(api): make panorama switch mode based on kwargs 2023-07-01 15:04:12 -05:00
Sean Sube db86108dc5
apply lint 2023-07-01 14:39:02 -05:00
Sean Sube 8dbecd6947
lint(api): avoid shadowing params var 2023-07-01 12:38:26 -05:00
Sean Sube 7736430fe7
fix(api): pass overlap params to tile calculation 2023-07-01 12:36:17 -05:00
Sean Sube f20afeb58f
fix(api): mark panorama outpainting tests as very slow, update refs 2023-07-01 12:33:21 -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 5e1b70091c
feat(api): remove size restrictions on most pipelines 2023-07-01 12:33:04 -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 fcaeb8cb42
fix(api): add non-square txt2img test 2023-07-01 12:32:22 -05:00
Sean Sube 5378619ef2
fix(api): add name filter and MSE multiplier args to release tests 2023-07-01 12:32:18 -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 7c8dc7a6b6
feat(api): create hash file for models without one 2023-06-27 08:17:21 -05:00
Sean Sube 16cf5bb863
fix(api): strip trailing newlines when reading hash from file 2023-06-27 07:33:23 -05:00
Sean Sube e3a458a736
feat(api): read model hash from file 2023-06-27 07:28:59 -05:00
Sean Sube 0029943c7d
fix(api): use diffusion model name without path when looking up hashes 2023-06-27 07:09:40 -05:00
Sean Sube 7cbb4197ba
fix(api): handle empty negative prompts in exif data 2023-06-26 21:38:55 -05:00
Sean Sube 732eb9913e
add hash field to extras schema and user guide 2023-06-26 19:18:20 -05:00
Sean Sube 062b1c47aa
feat(api): write model hashes to image exif 2023-06-26 17:24:34 -05:00
Sean Sube 003a350a6c
feat(api): include version number in logs 2023-06-26 17:24:03 -05:00
Sean Sube 01811abf5a
add loras to exif params 2023-06-26 07:48:39 -05:00
Sean Sube 14e920c6e1
add newlines to make params compatible 2023-06-26 07:24:25 -05:00
Sean Sube f142f83ad5
dump json parameters to str before writing to exif 2023-06-26 07:14:32 -05:00
Sean Sube 91c6c62159
remove unused output param 2023-06-26 07:06:31 -05:00
Sean Sube 7e21b9539e
feat(api): embed image parameters in EXIF data (#383) 2023-06-26 07:03:06 -05:00
Sean Sube 9d08676caa
fix(api): add version for omegaconf 2023-06-19 18:12:56 -05:00
Sean Sube 03402dba67
fix(api): pin pip dependencies, add requirements to renovate 2023-06-19 18:02:00 -05:00
Sean Sube 7db53a82df
fix(api): correctly handle sliced kernels 2023-06-17 14:51:58 -05:00
Sean Sube 1506f51ff4
fix(api): handle more SwinIR models 2023-06-16 21:53:49 -05:00
Sean Sube 12e489b761
fix(api): use correct tile size for highres upscaling 2023-06-16 21:22:00 -05:00
Sean Sube 5713957026
fix(api): keep kernel slices within bounds 2023-06-16 20:49:43 -05:00
Sean Sube c1a4484b75
apply lint 2023-06-16 20:43:28 -05:00
Sean Sube 81d2e93709
chore(scripts): update DPM test ref for fp16 2023-06-16 20:42:48 -05:00
Sean Sube 719b34967f
fix(api): handle blending of mismatched kernels 2023-06-16 20:41:25 -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 d47dab8182
fix(api): make LPW pipeline valid again 2023-06-11 07:39:29 -05:00
Sean Sube 82a80bb60b
fix(api): ensure gradient points are always in order 2023-06-10 15:20:17 -05:00
Sean Sube ac2eceb0ac
fix(api): remove trailing space from param name 2023-06-10 11:54:33 -05:00
Sean Sube e64b813ac0
fix(api): pass inpaint param to checkpoint extraction 2023-06-10 11:52:25 -05:00
Sean Sube 665cf57a86
fix typo 2023-06-10 11:30:47 -05:00
HoopyFreud 22ff0be758 fix inpainting conversion via torch 2023-06-10 12:01:02 -04:00
Sean Sube c399430274
fix(scripts): only mark release tests as failed after retrying 2023-06-09 17:41:21 -05:00
Sean Sube 732b07eb46
fix(api): skip gradient blending when tiles do not overlap 2023-06-09 17:24:08 -05:00
Sean Sube f4d944acbe
apply lint 2023-06-08 07:20:19 -05:00
Sean Sube 3f00da9681
feat(api): add option for CPU-only conversion on systems with CUDA 2023-06-08 07:20:03 -05:00
Sean Sube 8215a1bf8a
fix(api): do not overlap tiles by default 2023-06-08 07:09:10 -05:00
Sean Sube 55614cd9a1
fix(scripts): restore MSE calculation, update some test refs 2023-06-07 21:13:09 -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 78e606b049
lint(api): move some progress messages to info level 2023-06-05 23:47:32 -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 6a09404243
disable histogram matching for now 2023-06-03 21:00:59 -05:00
Sean Sube 7a00d5d446
fix overlap factor 2023-06-03 20:56:56 -05:00
Sean Sube 1abd7060dc
remove second uint conversion 2023-06-03 20:46:46 -05:00
Sean Sube e79a157957
blend in float, paste results back into scratch texture 2023-06-03 20:38:16 -05:00
Sean Sube 55e5f73d20
apply lint 2023-06-03 20:35:33 -05:00
Sean Sube ec8f367316
use overlap param, apply to spiral tiling as well 2023-06-03 20:35:21 -05:00
Sean Sube e64be674ea
crop count layer 2023-06-03 14:02:58 -05:00
Sean Sube 3bdf3692f4
crop output image 2023-06-03 13:50:27 -05:00
Sean Sube 20bf5ecbfc
use scaled dims for max 2023-06-03 13:28:09 -05:00
Sean Sube 51c5b63090
use adjusted tile for gradients, repeat mask based on output shape 2023-06-03 13:17:20 -05:00
Sean Sube ec4ee49806
use mask for count 2023-06-03 12:22:50 -05:00
Sean Sube 4810ca858a
fix(api): correct name of torch extraction var 2023-06-03 09:52:31 -05:00
Sean Sube a19bf06003
attempt gradient blending of tiles 2023-06-03 09:51:44 -05:00
Sean Sube 6317094c5f
feat(api): histogram equalization and blending for highres tiling 2023-06-02 01:14:40 -05:00
Sean Sube 4dc192a2c7
apply lint 2023-05-28 19:44:16 -05:00
Sean Sube 3e966096ce
fix(api): call venv during setup scripts (#378) 2023-05-28 19:43:21 -05:00
Sean Sube 00b1bec7a9
fix(api): skip loading custom VAE after torch extraction (#379) 2023-05-27 12:31:08 -05:00
Sean Sube 3adafeaa3d
fix(api): switch back to torch extraction for now (#379) 2023-05-27 12:29:17 -05:00
Sean Sube 86d458a0c6
fix(api): avoid passing cache path before sanitizing config name 2023-05-27 12:27:33 -05:00
Sean Sube 24f1ecc0eb
fix(api): load local custom VAE from within model path 2023-05-27 11:26:16 -05:00
Sean Sube 1562f4011e
replace pydantic polyfill 2023-05-20 20:19:41 -05:00
Sean Sube eb53238fd5
do not override source after fetch 2023-05-20 20:15:19 -05:00
Sean Sube df2e854dcc
fix load_yaml import 2023-05-20 20:09:45 -05:00
Sean Sube 9fa1f0387b
add env var for torch extraction 2023-05-20 20:06:54 -05:00
Sean Sube 9c28154dee
feat(api): add option to reload CNet for conversion 2023-05-20 15:19:38 -05:00
Sean Sube 20107f559a
use Torch working model for conversion 2023-05-20 15:12:39 -05:00
Sean Sube 339441cb16
fix(api): restore old checkpoint conversion code path 2023-05-20 13:41:23 -05:00
Sean Sube 44ad5150b2
fix(scripts): add test ref for img2img panorama 2023-05-20 13:38:47 -05:00
Sean Sube 5ede55499a
fix(api): add arpeggio parser to base requirements 2023-05-15 00:08:25 -05:00
Sean Sube fe80e61e79
fix CNet conversion condition 2023-05-14 20:53:12 -05:00
Sean Sube b6a4cbaae1
fix(api): collate CNet after unloading UNet 2023-05-14 20:43:16 -05:00
Sean Sube 01ab864a55
collate UNet after unloading original, add option to skip pipeline loading test 2023-05-14 20:29:18 -05:00
Sean Sube 56fa0729b5
fix conversion context 2023-05-14 20:05:49 -05:00
Sean Sube 0d51d61728
feat(api): add env vars for controlnet conversion and opset 2023-05-14 20:04:43 -05:00
Sean Sube a99388645a
correct import 2023-05-14 19:56:40 -05:00
Sean Sube 579c5f9e97
fix(api): scope patched imports and run GC after patching 2023-05-14 19:54:57 -05:00
Sean Sube f53e8c1761
feat(scripts): add release tests for panorama and upscaling with correction 2023-05-14 19:31:17 -05:00
Sean Sube e2035c3fbf
fix(api): run GC during diffusers conversion, add flag to skip ControlNet (#369) 2023-05-14 19:30:30 -05:00
Sean Sube e96bd0fca4
feat(api): add lock to conversion endpoint 2023-05-10 08:17:22 -05:00
Sean Sube cbc4166f10
fix(scripts): add more release tests 2023-05-09 21:51:17 -05:00
Sean Sube bbd779a9c6
feat(scripts): add retries to release tests 2023-05-09 20:55:48 -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 f1856cb0d2
reload models and extras after conversion 2023-05-08 21:59:27 -05:00
Sean Sube 6f38d35c73
default content type, use arg parser for override 2023-05-08 21:56:05 -05:00
Sean Sube c992c03584
write extras as string, override conversion args from endpoint 2023-05-08 21:49:36 -05:00
Sean Sube a03f89a62c
fix(api): add env var for admin token 2023-05-08 21:42:25 -05:00
Sean Sube 534fc70e29
implement conversion endpoint 2023-05-08 21:40:51 -05:00
Sean Sube d39c7920a2
feat(api): add endpoints to manage extras file 2023-05-06 16:49:02 -05:00
Sean Sube 0f1298824c
feat(api): start adding support for multiple prompt parsers 2023-05-06 11:41:57 -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 da6aa15afc
fix(api): fully qualify HF models (#371) 2023-05-05 18:33:54 -05:00
Sean Sube 88fa126d69
Merge branch 'feat/373-non-square-inpaint' 2023-05-03 23:07:17 -05:00
Sean Sube 23aaf656d3
fix(api): match mask and image size before adding noise (#373) 2023-05-03 23:05:40 -05:00
Sean Sube d66bf9e54f
fix(api): download pretrained models from HF correctly (#371) 2023-05-03 22:21:04 -05:00
Sean Sube b6692f068e
fix(api): parse config files as JSON or YAML depending on extension (#370) 2023-05-03 22:15:17 -05:00
Sean Sube e68e405633
fix: limit UNet stride to tile size 2023-05-03 19:21:32 -05:00
Sean Sube 3b02fc5768
feat: add UNet stride as its own parameter 2023-05-03 19:15:05 -05:00
Sean Sube ac4aef6ae3
resize mask to cover source image before trimming 2023-05-02 21:15:14 -05:00
Sean Sube c8bc175239
fix(api): keep aspect ratio of mask for non-square sources (#373) 2023-05-02 20:52:28 -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 ca611f03df
enable all valid inpaint pipes 2023-05-01 18:58:29 -05:00
Sean Sube 71b95182a1
feat(api): add inpaint to panorama pipeline 2023-05-01 18:53:46 -05:00
Sean Sube a57d90ca02
fix(api): add convertible pipelines to extras schema 2023-04-30 21:23:44 -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 852c57c08a
fix(api): add pytorch_lightning to optional reqs (#366) 2023-04-30 17:19:03 -05:00
Sean Sube 124014b8b5
fix(exe): copy entrypoint script to correct location before bundling 2023-04-30 14:56:24 -05:00
Sean Sube 572a5159ad
fix(api): limit fp16 ops for v2.1 models (#364) 2023-04-30 08:49:34 -05:00
Sean Sube ea9d15fdd2
chore(api): prepare log configs for detailed log levels 2023-04-30 07:35:09 -05:00
Sean Sube 52cc4baaa4
fix circular import 2023-04-29 23:31:08 -05:00
Sean Sube 0573721d12
fix(api): always use accumulating progress counters 2023-04-29 23:28:51 -05:00
Sean Sube bf2f77d7a2
apply lint 2023-04-29 23:12:15 -05:00
Sean Sube 077845e868
fix log levels 2023-04-29 23:11:46 -05:00
Sean Sube 62abc76f3f
convert checkpoint dtype after loading 2023-04-29 23:08:48 -05:00
Sean Sube e28bbe2c18
set safetensor and dtype when loading 2023-04-29 23:05:51 -05:00
Sean Sube 1c308e0bab
collect and summarize errors while converting models 2023-04-29 22:58:58 -05:00
Sean Sube bc71583393
fix(api): add model image size and version hint to extras file 2023-04-29 22:56:52 -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 4eba9a6400
avoid joining none path 2023-04-29 21:16:23 -05:00
Sean Sube 5760be710a
fix(api): restore use of config key when converting from checkpoints 2023-04-29 20:26:01 -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 5d5bfdbbcc
fix: increase max image sizes 2023-04-29 15:59:59 -05:00
Sean Sube b4dfc44bc6
fix(api): return correct output from tiled VAE 2023-04-29 15:59:39 -05:00
Sean Sube f420a3a612
feat: experimentally add TensorRT runtime 2023-04-29 15:55:58 -05:00
Sean Sube fbab26fe31
resize mask to match source 2023-04-29 15:40:26 -05:00
Sean Sube bfe989997e
use full tile size for mask and noise 2023-04-29 15:10:43 -05:00
Sean Sube a1e7a5d077
pad incomplete mask tiles 2023-04-29 15:00:59 -05:00
Sean Sube a4d197a6e7
round entire canvas up to tile size 2023-04-29 14:55:41 -05:00
Sean Sube f782f39cce
fix(api): remove size limit on inpainting stage 2023-04-29 14:23:00 -05:00
Sean Sube acde899559
reuse config from loaded UNet 2023-04-29 13:27:39 -05:00
Sean Sube 0175d7edcf
fix(api): specify input channels when converting inpainting models (#356) 2023-04-29 12:18:15 -05:00
Sean Sube 4c126153f5
fix(api): switch to diffusers ckpt loading, add more pipelines to conversion (#337, #356) 2023-04-29 12:00:43 -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 d9f7971d0a
fix(api): do not downscale source images for panorama pipeline 2023-04-28 16:02:03 -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 7e5380db43
feat: add tiled VAE parameter to client 2023-04-28 13:30:54 -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 dbd9a186ae
feat(api): allow converting inpaint models (#356) 2023-04-27 23:41:04 -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 b43cb26d06
fix(build): copy schemas into containers 2023-04-24 19:23:52 -05:00
Sean Sube dc6bf330c1
feat(api): add optimization for max attention slicing (#355) 2023-04-24 19:13:32 -05:00
Sean Sube 95e75ffd24
fix(api): check GTK import individually 2023-04-24 18:39:25 -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 ec7d563c58
fix(api): resolve SD upscaling model relative to models path 2023-04-24 18:23:56 -05:00
Sean Sube ae9f6c2f35
fix(api): add requirements files for optional deps 2023-04-24 18:13:00 -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 7caaa9ed40
fix(api): correct dict types for 3.8 2023-04-24 17:39:06 -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 9e9fe90f80
chore(api): remove SD v2.1 from base models 2023-04-23 18:23:05 -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 6fecb1ee8b
fix(api): get admin token from correct server object 2023-04-22 13:16:14 -05:00
Sean Sube 4ac25f0650
fix(api): remove traces of highres from blend pipeline 2023-04-22 12:40:51 -05:00
Sean Sube 5f5b01fed0
apply lint 2023-04-22 12:28:46 -05:00
Sean Sube e283b2770a
fix(scripts): do not use source filter for img2img tests 2023-04-22 12:28:13 -05:00
Sean Sube 5f5418132b
fix(api): build sum tokens for TIs using emb_params key 2023-04-22 12:07:52 -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 eeaba37fa9
fix(api): correctly disable source filtering when none is selected 2023-04-22 11:34:48 -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 c96dd814a8
fix(api): correct slashes in launch scripts 2023-04-22 10:39:39 -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 9bb612f54e
fix(api): handle empty prompts (#340) 2023-04-22 00:28:10 -05:00
Sean Sube 6e7f202b8d
fix(api): store both pre-parse and parsed prompts (#320) 2023-04-22 00:11:33 -05:00
HoopyFreud 21b53c1b7b Recomitting everything 2023-04-21 15:17:31 -04:00
Sean Sube c7aea34b28
fix(api): use correct params for default outpainting borders (#345) 2023-04-20 22:33:18 -05:00
Sean Sube a46f00aeb4
fix(api): use slice multiplication for 3x3 kernels without CP decomp 2023-04-20 22:10:49 -05:00
Sean Sube 49781df2fd
apply lint 2023-04-20 21:26:16 -05:00
Sean Sube ff4c8a9404
handle tensors of the same rank 2023-04-20 20:24:37 -05:00
Sean Sube 3e8f4b3edf
fix(api): improve summing of mismatched weights 2023-04-20 20:06:43 -05:00
Sean Sube f0109d3406
fix(api): blend LoHA and LoRA weights for 1x1 kernels 2023-04-20 19:31:40 -05:00
Sean Sube e9e9d75f9d
fix(api): correct return value after fully converting checkpoints 2023-04-20 18:00:35 -05:00
Sean Sube 30c96be24f
send token with admin requests, return worker status after restarting 2023-04-20 17:36:29 -05:00
Sean Sube df0e7dc57e
feat: add admin endpoint to restart image workers (#207) 2023-04-20 07:36:31 -05:00
Sean Sube e12f3c2801
fix(scripts): correct model paths in test scripts 2023-04-16 23:18:35 -05:00
Sean Sube cfdd926fff
fix(api): wait for worker to become idle before enqueueing next job (#286) 2023-04-15 20:37:53 -05:00
Sean Sube 17e7b6aff2
fix(api): keep ControlNet metadata after lookup 2023-04-15 17:31:02 -05:00
Sean Sube 8d47b71183
fix(api): match ControlNet selection against model names 2023-04-15 17:21:13 -05:00
Sean Sube 31965356d8
fix(api): handle mixed tensors for ControlNet on CPU 2023-04-15 17:03:04 -05:00
Sean Sube 161913bb7e
fix(api): add pre-converted ControlNets to base sources, add missing ControlNet strings 2023-04-15 16:55:53 -05:00
Sean Sube e0e6b74b98
fix(api): correct parameter names for BSRGAN, use RRDB from BasicSR for Real ESRGAN 2023-04-15 16:20:46 -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 1853214409
fix(build): install CPU deps during CI 2023-04-15 13:18:24 -05:00
Sean Sube f5c606e16f
remove old requirements file 2023-04-15 12:51:47 -05:00
Sean Sube 7b4ce44f51
fix(api): print border size in expected order 2023-04-15 12:36:41 -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 43d545e524
apply lint 2023-04-15 12:13:25 -05:00