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

1665 Commits

Author SHA1 Message Date
Sean Sube 9a1389d03b
fix imports 2024-01-04 19:16:40 -06:00
Sean Sube b6da935be6
use progress type in command 2024-01-04 19:09:52 -06:00
Sean Sube ce84dfa115
pass metadata through tiles 2024-01-04 09:12:32 -06:00
Sean Sube 112d5a8876
start passing metadata through stages 2024-01-04 09:01:19 -06:00
Sean Sube e3a6635ca2
remove unused job type field 2024-01-03 23:58:57 -06:00
Sean Sube 48671e27b0
fix hash field 2024-01-03 23:46:50 -06:00
Sean Sube fe6b267483
prep models field 2024-01-03 23:38:44 -06:00
Sean Sube 04be627e7b
pass server when serializing metadata 2024-01-03 23:32:33 -06:00
Sean Sube bdefaec0d2
serialize metadata 2024-01-03 23:27:40 -06:00
Sean Sube 8ae6069ac3
handle 0 progress 2024-01-03 23:19:56 -06:00
Sean Sube 7a5e6ab895
pass result on to command 2024-01-03 23:17:35 -06:00
Sean Sube fac98ab239
keep result of each stage with metadata 2024-01-03 23:13:21 -06:00
Sean Sube e0d0933092
fix missing param 2024-01-03 22:20:27 -06:00
Sean Sube 46bcd5af86
make response types match 2024-01-03 22:15:50 -06:00
Sean Sube c4b831fe5c
fix typos 2024-01-03 21:45:13 -06:00
Sean Sube 9b5e894898
avoid waiting for final progress 2024-01-03 21:39:19 -06:00
Sean Sube 28fc2082c7
track results after each stage 2024-01-03 21:31:41 -06:00
Sean Sube 4f1bc84fd9
initialize fields 2024-01-03 21:20:46 -06:00
Sean Sube 0c504e3f69
report stage and tile count in progress 2024-01-03 21:16:44 -06:00
Sean Sube 9d05d9baac
return output names 2024-01-03 20:54:11 -06:00
Sean Sube 492bc1e992
make sure image response is array 2024-01-03 20:28:53 -06:00
Sean Sube d87861cb29
fix params, save metadata 2024-01-03 20:22:38 -06:00
Sean Sube 444d809eff
fix progress, unwrap metadata 2024-01-03 19:18:56 -06:00
Sean Sube a6baf369d4
avoid reusing routes 2024-01-03 19:15:33 -06:00
Sean Sube 44a8d61082
feat: add batch endpoints for cancel and status, update responses 2024-01-03 19:09:18 -06:00
Sean Sube 19c91f70f5
add networks to metadata 2024-01-02 22:14:21 -06:00
Sean Sube 46098960d8
attempt to save grid mode metadata 2024-01-02 21:49:22 -06:00
Sean Sube 297285fb6f
fix(api): upscale during highres regardless of upscale checkbox 2024-01-02 21:34:11 -06:00
Sean Sube 4edd39740b
add image metadata to stage results 2024-01-02 21:24:27 -06:00
Sean Sube 0fa7eff8a8
make thumbnail size configurable 2024-01-02 21:24:09 -06:00
Sean Sube 5ad7b4c1b4
fix thumbnail call 2024-01-02 21:19:34 -06:00
Sean Sube 5de28d0361
generate a thumbnail when the first image is > 512 2024-01-02 20:58:02 -06:00
Sean Sube 35371d33fe
chore(release): 0.12.0 2023-12-31 14:15:31 -06:00
Sean Sube 00db9a2787
fix(api): make sure ControlNet constructor is used for same pipeline (#449) 2023-12-31 13:00:34 -06:00
Sean Sube de8ab6f1c9
fix(api): use correct part of the gfpgan output 2023-12-31 07:04:43 -06:00
Sean Sube c6718a3e3b
fix(api): make sure upscaling only runs once regardless of correction options 2023-12-31 06:39:00 -06:00
Sean Sube 71fbc871a6
feat: add API parameter for upscale checkbox 2023-12-30 19:34:40 -06:00
Sean Sube 54db63394f
arch adjustments 2023-12-30 13:50:14 -06:00
Sean Sube 366efd1a58
fix RRDB rescale arch 2023-12-30 13:34:12 -06:00
Sean Sube 6834b716ea
switch RRDB nets based on upscaling 2023-12-30 13:28:16 -06:00
Sean Sube 0ddc16288f
feat(api): support both ESRGAN variants 2023-12-30 13:11:56 -06:00
Sean Sube 6cee411e43
add ESRGAN input adjustments back 2023-12-30 12:42:50 -06:00
Sean Sube 7abe6dc6a9
feat(api): enable 1x upscaling models 2023-12-30 11:59:52 -06:00
Sean Sube 11e643bcb5
remove broken channel adjustment 2023-12-30 11:59:37 -06:00
Sean Sube 477747cced
better support for ESRGAN 1x models 2023-12-30 11:50:28 -06:00
Sean Sube f6134dafc6
fix errors in local std denoising 2023-12-29 23:19:35 -06:00
Sean Sube 6a004816af
fix(api): make version-safe imports compatible with tests 2023-12-29 23:09:00 -06:00
Sean Sube 05ab396b2a
revert back to NL means for panorama only 2023-12-29 22:53:04 -06:00
Sean Sube 750a49a907
apply lint 2023-12-29 18:53:13 -06:00
Sean Sube 8b698ed4ab
Revert "wrap some diffusers imports for version safety" 
This reverts commit e7f7bbc9ea.
 2023-12-29 18:52:54 -06:00
Sean Sube da6beb31b2
switch to internal RRDBNet with x2 support 2023-12-29 08:50:27 -06:00
Sean Sube 1035915d36
apply lint 2023-12-29 08:19:58 -06:00
Sean Sube ce90ffb0ee
use threshold range for local denoising 2023-12-29 08:18:37 -06:00
Sean Sube 973677967d
fix(api): turn off SDXL's very visible watermarking (#438) 2023-12-28 22:54:53 -06:00
Sean Sube e7f7bbc9ea
wrap some diffusers imports for version safety 2023-12-28 22:49:17 -06:00
Sean Sube f28fdda47a
feat(api): add stage for local standard deviation denoising for XL 2023-12-27 20:17:35 -06:00
Sean Sube 7d56689527
pass device to wrapper 2023-12-27 09:01:38 -06:00
Sean Sube d17b946091
wrap ESRGAN helper 2023-12-27 08:47:06 -06:00
Sean Sube 404f24f9ad
compile pattern 2023-12-27 05:17:17 -06:00
Sean Sube 95886430a4
feat(api): support more RealESRGAN-based models 2023-12-27 05:08:15 -06:00
Sean Sube e8a30ae56b
feat(api): add optimization flag to ORT extended optimizations 2023-12-26 20:22:32 -06:00
Sean Sube 006f983269
fix(api): use standard op block list for SDXL 2023-12-26 20:22:09 -06:00
Sean Sube cfe7a55935
type and test fixes 2023-12-26 20:21:34 -06:00
Sean Sube eb8bd145c9
null checks 2023-12-26 08:07:05 -06:00
Sean Sube 2b65077d82
type lint 2023-12-26 08:06:16 -06:00
Sean Sube ae0e8447f9
add message to unet type error 2023-12-24 23:10:08 -06:00
Sean Sube e2c9389d6e
add fallback dtypes to unet patch 2023-12-24 22:57:02 -06:00
Sean Sube ef256280b4
use right type mapping 2023-12-24 22:46:22 -06:00
Sean Sube 0f6a1a82a2
translate types to np 2023-12-24 22:40:07 -06:00
Sean Sube 39ee4cbfcd
handle XL UNets 2023-12-24 22:36:39 -06:00
Sean Sube 80a255397e
feat(api): use wrapped model's input types in UNet patch 2023-12-24 22:21:52 -06:00
Sean Sube 4eeb68cbcf
fix(api): allow numbers in archive filenames 2023-12-24 12:07:18 -06:00
Sean Sube 15b9b51ed5
fix(api): make sure new SD converter does not produce extra files 2023-12-24 10:47:46 -06:00
Sean Sube 29ccd16437
fix(api): handle HF repos in new SD converter 2023-12-24 07:05:56 -06:00
Sean Sube e707bc8dde
re-enable torch extraction by default in legacy converter 2023-12-24 06:25:24 -06:00
Sean Sube da6e813638
rename old converter 2023-12-24 06:24:45 -06:00
Sean Sube 5d31e85709
use torch suffix consistently 2023-12-24 06:16:37 -06:00
Sean Sube 48630b3dfe
add torch suffix to temp dir 2023-12-24 06:12:43 -06:00
Sean Sube 525fec24c5
go back to working dir 2023-12-24 06:10:15 -06:00
Sean Sube 769350115c
pass correct paths to torch extraction 2023-12-24 06:00:44 -06:00
Sean Sube 9ffe266384
fix(api): unload UNet before CNet conversion 2023-12-24 05:32:39 -06:00
Sean Sube 054e5c406a
fully rename old converter to legacy 2023-12-24 05:31:55 -06:00
Sean Sube 9a077fae15
feat(api): replace legacy SD converter with optimum-based one by default 2023-12-23 22:55:52 -06:00
Sean Sube a5a60d3e15
fix(api): GC during SDXL conversion 2023-12-23 22:34:05 -06:00
Sean Sube f6e6e31789
add controlnet support to optimum converter 2023-12-23 22:33:46 -06:00
Sean Sube abeeddeeb2
fix(api): handle models with integer timestamps 2023-12-23 22:19:15 -06:00
Sean Sube ebe813d035
fix(api): change weights filename for new models to match optimum 2023-12-23 22:18:35 -06:00
Sean Sube 14208de393
register optimum-based SD converter 2023-12-23 22:18:09 -06:00
Sean Sube b6ef00e437
feat(api): add new optimum-based SD converter 2023-12-23 22:09:57 -06:00
Sean Sube 2b8b59a39c
make SD conversion more like SDXL 2023-12-23 21:42:36 -06:00
Sean Sube 50d51a3921
fix(api): make argument for skipping base models work correctly 2023-12-23 14:21:39 -06:00
Sean Sube 52065ef317
lint(api): remove some confusing log messages 2023-12-23 12:17:23 -06:00
Sean Sube 3d245a663d
fix(api): correct origin point when blending images (#445) 2023-12-22 11:31:31 -06:00
Sean Sube bc9276c253
lint imports 2023-12-21 21:34:03 -06:00
Sean Sube ccff9a3136
fix(api): silence controlnet registration warnings 2023-12-21 21:30:49 -06:00
Sean Sube 2546084208
lint(api): move some noisy log messages to trace level 2023-12-21 21:07:02 -06:00
Sean Sube 6bec3b0cb9
fix(api): change default web UI path to match docs 2023-12-21 20:41:52 -06:00
Sean Sube 2c2eda8c3c
fix(api): resize blend mask to match source 2023-12-20 23:41:54 -06:00
Sean Sube 1fbee0ae52
fix(api): tile stage masks 2023-12-20 23:33:13 -06:00
Sean Sube f6ff53d0d1
pass correct crop 2023-12-20 23:02:58 -06:00
Sean Sube de28ecdfad
fix(api): handle blending images of different sizes better 2023-12-20 22:58:57 -06:00
Sean Sube 9a3384b348
Merge branch 'main' of github.com:ssube/onnx-web 2023-12-20 09:21:56 -06:00
Sean Sube 9c215dd90b
fix controlnet with fp16 again 2023-12-19 23:16:19 -06:00
Sean Sube e09631ef05
apply lint 2023-12-19 23:01:05 -06:00
Sean Sube c8a9734acf
dump unet to torch before making cnet 2023-12-19 22:50:18 -06:00
Sean Sube 4a87fb2a31
update forwarding shapes, apply lint 2023-12-19 22:10:37 -06:00
Sean Sube b4262eb777
remove bulky controlnet wrapper 2023-12-19 20:56:24 -06:00
Sean Sube ba9982af51
fix(api): various controlnet fixes 2023-12-19 19:19:48 -06:00
Sean Sube a716f6db1f
fix(api): add loading path for controlnet 2023-12-18 22:21:33 -06:00
Sean Sube 6708efa8ce
make controlnet params match others 2023-12-18 22:14:33 -06:00
Sean Sube 6e5d10f521
apply lint 2023-12-18 22:04:24 -06:00
Sean Sube 2263093c4d
convert images to cv2 2023-12-18 21:53:20 -06:00
Sean Sube 1163accb49
fix(api): add missing param to controlnet pipeline 2023-12-18 21:39:49 -06:00
Sean Sube 3e1db707ac
fix(api): update codeformer patches for new lib 2023-12-18 21:12:08 -06:00
Sean Sube 7ed30ee470
feat(api): switch to codeformer lib that works with torch 2.x 2023-12-18 21:06:39 -06:00
BZ Libby fc67809612
clean-up 
I missed removing a character in the regex for the wildcard token
 2023-12-18 19:09:23 -06:00
BZLibby b807530117 fixes support for multiple wildcards in one token 2023-12-18 19:01:18 -06:00
Sean Sube 9e734706f1
fix(api): run blend pipeline in RGB mode (#439) 2023-12-17 14:54:10 -06:00
Sean Sube 76e28d962d
fix(api): parse embedding/TI tokens correctly 2023-12-17 12:48:20 -06:00
Sean Sube 41b8ae0359
open archives in read-only mode 2023-12-17 12:06:43 -06:00
Sean Sube a65e0fd602
update schemas for archive converter 2023-12-16 22:49:29 -06:00
Sean Sube f36fa6b5ec
bump server version in logs 2023-12-16 22:40:04 -06:00
Sean Sube f3e1beaa71
feat(api): add converter to extract ZIP archives (#437) 2023-12-16 22:14:43 -06:00
Sean Sube 2653dd812e
feat(api): allow embeddings as a synonym for TI token 2023-12-16 15:27:43 -06:00
Sean Sube 02dd17aa4f
fix(api): allow spaces in wildcard filenames 2023-12-16 15:21:49 -06:00
Sean Sube cdbdd9b4e2
feat(gui): add wildcard menu to web UI 2023-12-16 15:17:28 -06:00
BZLibby 90bc28d06e apply style 2023-12-16 14:56:19 -06:00
BZLibby 6a561173a7 adds support for dynamic wildcards in prompts 2023-12-16 14:37:56 -06:00
Sean Sube 133c4a20bd
fix(api): use correct post-fetch path when converting from checkpoints (#432) 2023-12-16 13:17:24 -06:00
Sean Sube 695eeaf303
feat(api): remove deprecated Karras Ve scheduler (#189) 2023-12-14 22:36:52 -06:00
Sean Sube 6c4f4f334f
always use alpha in blend stage 2023-12-14 20:12:39 -06:00
Sean Sube 95d8f4a598
fix(api): make sure all file types are covered (#432) 2023-12-14 20:04:08 -06:00
Sean Sube 4da4cd95a5
fix(api): trim whitespace from model names because it breaks things (#376) 2023-12-10 13:59:47 -06:00
Sean Sube 9c1fcd16fa
fix(api): only fetch diffusion models if they have not already been converted (#398) 2023-12-10 13:52:52 -06:00
Sean Sube c9b1df9fdd
use dest in file client if provided 2023-12-10 13:30:27 -06:00
Sean Sube e91e08484b
handle query params better in civitai client 2023-12-10 13:17:37 -06:00
Sean Sube e200fe9186
avoid pickling environ 2023-12-10 13:07:02 -06:00
Sean Sube 9b883de1cb
pass context to client ctor 2023-12-10 12:16:01 -06:00
Sean Sube 3b32cd4ac3
sonar lint 2023-12-10 11:45:52 -06:00
Sean Sube 4edb32aaac
feat(api): add env vars for Civitai and Huggingface auth tokens 2023-12-10 11:08:02 -06:00
Sean Sube e6978243cc
fix download location for networks 2023-12-10 10:57:25 -06:00
Sean Sube d9c5c1bd45
lint 2023-12-09 23:51:43 -06:00
Sean Sube 2fc5ec930c
feat(api): add support for DPM SDE scheduler 2023-12-09 23:51:09 -06:00
Sean Sube 50db19922a
fix type of client instance 2023-12-09 23:40:47 -06:00
Sean Sube 22e0597916
pass kwargs on to client 2023-12-09 23:09:00 -06:00
Sean Sube 35c973e55f
pass correct metadata to network converter 2023-12-09 23:07:56 -06:00
Sean Sube 52e78748c8
keep model field optional 2023-12-09 23:05:41 -06:00
Sean Sube e052578a20
lint, give civitai client a name 2023-12-09 23:03:41 -06:00
Sean Sube ebb5a586ce
get file root path from context, avoid downloading entire HF repos 2023-12-09 22:53:29 -06:00
Sean Sube 6cdba4cebb
handle extensions correctly 2023-12-09 21:05:16 -06:00