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

356 Commits

Author SHA1 Message Date
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 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
Sean Sube 8bb76f1fee
keep protocol when downloading from http sources 2023-12-09 20:50:41 -06:00
Sean Sube 4c5bb906e8
pass conversion context to generic sources 2023-12-09 20:36:04 -06:00
Sean Sube 20b719fdff
special downloading for embeds 2023-12-09 19:15:28 -06:00
Sean Sube 7496613f4e
fix fetch calls 2023-12-09 18:46:47 -06:00
Sean Sube cc8e564d26
split up download clients, add to plugin exports 2023-12-09 18:04:34 -06:00
Sean Sube e7da2cf8a6
fix(api): load pretrained VAE from original path 2023-12-08 23:19:52 -06:00
Sean Sube 4fd50984f0
fix(api): correct VAE extension check during conversion 2023-12-08 22:55:51 -06:00
Sean Sube 293a1bb184
fix(api): include SD upscaling in diffusion prefixes 2023-12-08 22:26:19 -06:00
Sean Sube 46d9fc0dd4
fix(api): make sure diffusion models have a valid prefix 2023-12-08 18:49:18 -06:00
Sean Sube b29837d773
fix(api): convert SD models without extracting Torch intermediate (#404) 2023-12-03 15:52:36 -06:00
Sean Sube 9b4ae0916b
lint and test fixes 2023-12-03 12:53:50 -06:00
Sean Sube 44a23a6366
apply lint 2023-11-24 22:40:01 -06:00
Sean Sube d7c95a4a4f
fix SDXL node names once 2023-11-24 16:51:03 -06:00
Sean Sube 74832fc61b
fix(api): match SDXL keys per LoRA 2023-11-24 15:22:07 -06:00
Sean Sube 98f8abbacd
fix(api): correctly detect unknown embedding format 2023-11-23 23:53:17 -06:00
Sean Sube ddd3d45b79
fix(api): add logging along SDXL custom VAE codepath 2023-11-23 23:21:45 -06:00
Sean Sube f00bfe9bd0
more tests, apply lint 2023-11-23 11:19:58 -06:00
Sean Sube 54dc970a7e
correctly remove operator types 2023-11-21 23:12:24 -06:00
Sean Sube 33bd67beb6
switch LoRA back to fixing node names on the fly 2023-11-21 22:33:17 -06:00
Sean Sube a39fe1d21c
use pre-fixed names for XL LoRA key matching 2023-11-21 21:46:34 -06:00
Sean Sube 60615e097b
fix(api): support normal LoRA keys in SDXL LoRAs 2023-11-21 19:10:11 -06:00
Sean Sube 7e6749e0d7
apply lint, add missing file 2023-11-18 18:13:13 -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 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 7c67d595fb
add more misc logging 2023-11-06 08:48:35 -06:00
Sean Sube b365660adc
move more SDXL LoRA logs to trace level 2023-11-05 16:35:32 -06:00
Sean Sube 046803fc2b
lint(api): reduce log level during SDXL LoRA blending 2023-11-05 15:55:49 -06: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 0ecae65f88
fix(api): set VAE attn processor during conversion 2023-09-24 15:02:21 -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 52fdf4f48a
fix(api): test LoRA blending code 2023-09-21 18:24:08 -05:00
Sean Sube a71298ff33
remove unused schema, lint 2023-09-15 08:40:56 -05:00
Sean Sube b851c234fe
more tests 2023-09-14 19:35:48 -05:00
Sean Sube 8a5e211172
lint(api): deduplicate some imports 2023-09-13 17:27:05 -05:00
Sean Sube cf2cf51b17
fix(api): resolve XL VAE within model folder 2023-09-11 20:21:43 -05:00
Sean Sube cd06f9291b
add custom VAE and fp16 support to SDXL conversion 2023-09-11 18:18:38 -05:00
Sean Sube 51b10de265
pass initial device to chain 2023-09-10 22:46:06 -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
Sean Sube fe68670844
feat(api): add conversion for SDXL models 2023-09-10 11:52:46 -05:00
Sean Sube 1d5ac7dde5
optimize XL LoRA node matching 2023-09-04 21:20:54 -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 404a314050
lint(api): add typed errors from cancelled job and download error 2023-09-04 21:20:04 -05:00
Sean Sube 8e1f188d8f
fix(api): pass replacement VAE to diffusers conversion 2023-09-04 21:19:36 -05:00
Sean Sube 6ec7777f77
lint(api): type fixes and hints throughout 2023-07-04 10:20:28 -05:00
Sean Sube 7c8dc7a6b6
feat(api): create hash file for models without one 2023-06-27 08:17:21 -05:00
Sean Sube 062b1c47aa
feat(api): write model hashes to image exif 2023-06-26 17:24:34 -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 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 719b34967f
fix(api): handle blending of mismatched kernels 2023-06-16 20:41:25 -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 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 4810ca858a
fix(api): correct name of torch extraction var 2023-06-03 09:52:31 -05:00
Sean Sube 4dc192a2c7
apply lint 2023-05-28 19:44:16 -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 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 e2035c3fbf
fix(api): run GC during diffusers conversion, add flag to skip ControlNet (#369) 2023-05-14 19:30:30 -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 da6aa15afc
fix(api): fully qualify HF models (#371) 2023-05-05 18:33:54 -05:00