Sean Sube
|
947a1bfdec
|
fix variable names
|
2023-02-25 11:26:10 -06:00 |
Sean Sube
|
44393e3770
|
feat(api): add txt2txt endpoint
|
2023-02-25 11:20:36 -06:00 |
Sean Sube
|
fd013c88ef
|
fix(api): enable optimizations during model conversion
|
2023-02-23 18:43:49 -06:00 |
Sean Sube
|
14c7e6a78c
|
apply lint
|
2023-02-21 08:26:09 -06:00 |
Sean Sube
|
82016a163a
|
restore single output in chain pipeline stages
|
2023-02-21 08:14:04 -06:00 |
Sean Sube
|
33ab23a474
|
convert to list of outputs
|
2023-02-20 23:47:43 -06:00 |
Sean Sube
|
b65e4e7cbe
|
use batch size when generating latents
|
2023-02-20 21:57:46 -06:00 |
Sean Sube
|
2c8bd4b043
|
read batch parameter from request
|
2023-02-20 21:46:23 -06:00 |
Sean Sube
|
5f3b84827b
|
feat(api): add batch size to txt2img and img2img pipelines (#195)
|
2023-02-20 08:35:18 -06:00 |
Sean Sube
|
0deaa8898d
|
fix(api): add missing params to SD upscale
|
2023-02-19 23:30:48 -06:00 |
Sean Sube
|
c1189aad96
|
feat: add eta parameter (fixes #194)
|
2023-02-19 23:29:26 -06:00 |
Sean Sube
|
40e396a2c4
|
sonar lint fixes
|
2023-02-19 22:10:35 -06:00 |
Sean Sube
|
f561dfae83
|
fix(api): correct blending mask (fixes #188)
|
2023-02-19 11:04:11 -06:00 |
Sean Sube
|
c459e96523
|
fix(api): report accurate size when face correction is run twice
|
2023-02-19 10:27:05 -06:00 |
Sean Sube
|
f7eefe4bd2
|
more sonar lint
|
2023-02-19 07:54:27 -06:00 |
Sean Sube
|
c69e342ee4
|
apply sonar lint
|
2023-02-19 07:53:20 -06:00 |
Sean Sube
|
30978e3e5b
|
fix(api): only run GC is devices are passed
|
2023-02-19 07:41:16 -06:00 |
Sean Sube
|
3789862a6f
|
mark unused params, fix shared lists
|
2023-02-19 07:37:29 -06:00 |
Sean Sube
|
b7a780bacc
|
correct param name for upscale mask
|
2023-02-18 22:53:57 -06:00 |
Sean Sube
|
30f16035bc
|
fix upscale copy ctor
|
2023-02-18 22:36:26 -06:00 |
Sean Sube
|
ac1f7449bb
|
fix(api): use stage source when available
|
2023-02-18 22:11:44 -06:00 |
Sean Sube
|
25c56c7d5c
|
fix imports
|
2023-02-18 20:28:21 -06:00 |
Sean Sube
|
3fd90a667d
|
lint(api): remove unused strength param from inpaint pipeline
|
2023-02-18 18:56:56 -06:00 |
Sean Sube
|
1a45aa68fa
|
clean up some exports, lint
|
2023-02-18 18:55:42 -06:00 |
Sean Sube
|
2b83f942af
|
restore original LPW names
|
2023-02-18 18:55:02 -06:00 |
Sean Sube
|
4d93c13431
|
apply lint, fix shadowed names
|
2023-02-18 17:59:13 -06:00 |
Sean Sube
|
b4f7973c1e
|
lint(api): use consistent name for source/mask images to avoid conflict with kwargs
|
2023-02-18 16:35:57 -06:00 |
Sean Sube
|
7b8ced0f68
|
feat(api): prefer chain stage parameters over request parameters (#138)
|
2023-02-18 16:27:48 -06:00 |
Sean Sube
|
bfdb071c2d
|
chore(docs): explain model optimizations
|
2023-02-18 16:06:05 -06:00 |
Sean Sube
|
e0a62ccbb5
|
better ONNX optimization logging
|
2023-02-18 15:47:31 -06:00 |
Sean Sube
|
881b290116
|
return session options properly
|
2023-02-18 15:45:28 -06:00 |
Sean Sube
|
5b4c370a1b
|
feat(api): enable ONNX optimizations through env
|
2023-02-18 15:44:39 -06:00 |
Sean Sube
|
0d2211ff25
|
apply lint
|
2023-02-18 14:14:13 -06:00 |
Sean Sube
|
f534fbb92c
|
fix(api): restore separate upscale and correction stages
|
2023-02-18 11:59:39 -06:00 |
Sean Sube
|
118695d68c
|
fix(api): add error handling for optimizations
|
2023-02-18 11:57:18 -06:00 |
Sean Sube
|
ab6462d095
|
feat(api): enable optimizations for SD pipelines based on env vars (#155)
|
2023-02-18 11:53:13 -06:00 |
Sean Sube
|
ff57527274
|
remove unused enum
|
2023-02-18 11:16:33 -06:00 |
Sean Sube
|
ffc4eb09f1
|
update reported image size when correction is run twice
|
2023-02-18 11:14:34 -06:00 |
Sean Sube
|
6dd0edc374
|
apply lint
|
2023-02-18 11:05:38 -06:00 |
Sean Sube
|
f051af359d
|
make default order match type
|
2023-02-18 11:05:23 -06:00 |
Sean Sube
|
ecf3c03f0f
|
feat: add parameter to run correction before upscaling (#132)
|
2023-02-18 11:04:08 -06:00 |
Sean Sube
|
b2de114c8c
|
fix(api): disable progress in diffusers pipelines
|
2023-02-18 09:42:38 -06:00 |
Sean Sube
|
8dd55cc556
|
fix(api): disable HF hub progress
|
2023-02-18 09:37:27 -06:00 |
Sean Sube
|
15060e6c7d
|
fix(api): create conversion context from environment vars
|
2023-02-18 09:31:20 -06:00 |
Sean Sube
|
b4e66ef502
|
feat(api): add flag to disable progress bars (#158)
|
2023-02-18 09:26:13 -06:00 |
Sean Sube
|
400e579491
|
Merge branch 'feat/113-onnx-resrgan'
|
2023-02-18 09:23:36 -06:00 |
Sean Sube
|
b3b10b4746
|
apply lint
|
2023-02-18 08:46:46 -06:00 |
Sean Sube
|
338fc237c7
|
fix(api): convert Real ESRGAN v3 using same arch as runtime
|
2023-02-18 08:41:15 -06:00 |
Sean Sube
|
431db6e3f8
|
repack size into tuples
|
2023-02-18 05:47:34 -06:00 |
Sean Sube
|
3ca02d4875
|
fix(api): make size params iterable
|
2023-02-18 05:44:43 -06:00 |
Sean Sube
|
0e108daa0f
|
fix(api): resize images to min dimensions by padding if necessary (#172)
|
2023-02-18 05:35:53 -06:00 |
Sean Sube
|
2c9d96d2ee
|
feat(api): use ONNX for Real ESRGAN v3 model
|
2023-02-17 23:25:42 -06:00 |
Sean Sube
|
ce8c7205dc
|
apply lint
|
2023-02-17 22:49:13 -06:00 |
Sean Sube
|
d09446ca68
|
fix(api): log number of active threads when running GC (#170)
|
2023-02-17 21:56:03 -06:00 |
Sean Sube
|
cf8ca34845
|
fix(api): correct device for VRAM check
|
2023-02-17 08:44:42 -06:00 |
Sean Sube
|
56a826228d
|
fix(api): only remove converted VAE if it exists
|
2023-02-17 08:34:28 -06:00 |
Sean Sube
|
7a1f8318ef
|
fix(api): only convert VAE once when coming from checkpoints
|
2023-02-17 08:29:23 -06:00 |
Sean Sube
|
37b173d0d1
|
fix(api): unwrap state dict from VAE
|
2023-02-17 08:23:12 -06:00 |
Sean Sube
|
c74d22aa42
|
fix(api): continue converting other models after an error in one (#166)
|
2023-02-17 07:49:45 -06:00 |
Sean Sube
|
b3c8fce16b
|
fix(api): fallback to PyTorch if tensors fail to load with JIT
|
2023-02-17 07:39:48 -06:00 |
Sean Sube
|
005650a9a2
|
feat(api): log CUDA memory stats (#141)
|
2023-02-16 23:51:17 -06:00 |
Sean Sube
|
d42de16a84
|
feat(api): support custom VAE for diffusers models
|
2023-02-16 22:52:25 -06:00 |
Sean Sube
|
388eb640c0
|
apply lint, rename lookup table
|
2023-02-16 22:22:46 -06:00 |
Sean Sube
|
f5ae9dd492
|
add ORT type table
|
2023-02-16 21:50:19 -06:00 |
Sean Sube
|
3d73b9e621
|
fix(api): update SD upscaling pipeline
|
2023-02-16 21:44:33 -06:00 |
Sean Sube
|
873276f1d0
|
load checkpoint properly
|
2023-02-16 20:23:10 -06:00 |
Sean Sube
|
ca1b22d44d
|
fix(api): support loading VAE from CKPT files
|
2023-02-16 20:18:42 -06:00 |
Sean Sube
|
4b6be765a6
|
copy VAE without translation
|
2023-02-16 19:24:02 -06:00 |
Sean Sube
|
cddbc87ca3
|
skip key removal for VAE tensors
|
2023-02-16 19:17:00 -06:00 |
Sean Sube
|
aa6dc334d7
|
resolve VAE files within model path
|
2023-02-16 18:56:54 -06:00 |
Sean Sube
|
2c66cc58c5
|
feat(api): add option for custom VAE to extras file (#145)
|
2023-02-16 18:53:50 -06:00 |
Sean Sube
|
5e9dfa3465
|
lint(api): lowercase log messages
|
2023-02-16 18:42:05 -06:00 |
Sean Sube
|
0ed4af18ad
|
fix(api): run torch gc alongside python (#156)
|
2023-02-16 18:11:35 -06:00 |
Sean Sube
|
6a2bac1359
|
lint(api): remove unused import
|
2023-02-15 21:01:25 -06:00 |
Sean Sube
|
841b18576c
|
apply lint
|
2023-02-15 20:50:12 -06:00 |
Sean Sube
|
d60ec52186
|
fix(api): report correct image size for blend tab
|
2023-02-15 18:15:35 -06:00 |
Sean Sube
|
32a4bbcd64
|
fix(api): remove alpha channel after blending but before upscaling (fixes #148)
|
2023-02-15 17:49:42 -06:00 |
Sean Sube
|
b5eeb44c62
|
fix(api): remove inpaint post-multiply (#150)
|
2023-02-15 17:45:25 -06:00 |
Sean Sube
|
df5fb93bc0
|
fix(api): bypass model cache entirely when limit is 0
|
2023-02-15 17:17:28 -06:00 |
Sean Sube
|
2b29b099f0
|
fix(api): make SD upscaling compatible with more schedulers
|
2023-02-15 17:16:20 -06:00 |
Sean Sube
|
3e5edb1c39
|
fix(api): restore python 3.8 compatibility (#146)
|
2023-02-14 21:23:16 -06:00 |
Sean Sube
|
4d0cd2e981
|
apply lint
|
2023-02-14 19:01:14 -06:00 |
Sean Sube
|
8a2a9174ba
|
fix(api): pass both device and session options to ORT (#38)
|
2023-02-14 18:57:50 -06:00 |
Sean Sube
|
feb4603171
|
fix(api): remove finished jobs from worker pool (#124)
|
2023-02-14 17:23:23 -06:00 |
Sean Sube
|
38f8aa38ee
|
lint(api): read blending sources in loop
|
2023-02-14 17:13:29 -06:00 |
Sean Sube
|
d473a0fd2d
|
fix(api): pass device options to ORT session (#38)
|
2023-02-14 17:12:52 -06:00 |
Sean Sube
|
f2d0025545
|
feat: add iPNDM scheduler
|
2023-02-14 07:27:51 -06:00 |
Sean Sube
|
1289c560bd
|
apply lint
|
2023-02-14 00:12:07 -06:00 |
Sean Sube
|
7f6fa22b4e
|
fix(api): skip model download if final converted version already exists (fixes #139)
|
2023-02-13 23:03:51 -06:00 |
Sean Sube
|
4cc1d63cec
|
fix(api): make diffusion model config optional again
|
2023-02-13 22:49:50 -06:00 |
Sean Sube
|
6b6f2f0eff
|
feat: add support for DEIS multistep scheduler
|
2023-02-13 22:37:54 -06:00 |
Sean Sube
|
05756b2b08
|
fix(api): patch more download paths (#134)
|
2023-02-13 22:23:17 -06:00 |
Sean Sube
|
d1b250660e
|
fix(api): convert blend sources to the same size as the mask
|
2023-02-13 21:57:47 -06:00 |
Sean Sube
|
c61f200bc2
|
fix(api): base model cache size on number of workers
|
2023-02-13 18:46:33 -06:00 |
Sean Sube
|
c432ab0795
|
remove oldest items from model cache first
|
2023-02-13 18:33:06 -06:00 |
Sean Sube
|
c6b2751cc1
|
fix(api): patch download fn in facexlib (#134)
|
2023-02-13 18:30:09 -06:00 |
Sean Sube
|
6d503ca00e
|
update cache entries with tag and key
|
2023-02-13 18:27:20 -06:00 |
Sean Sube
|
51d53597c0
|
fix(api): load Real ESRGAN PTH from cache
|
2023-02-13 18:21:05 -06:00 |
Sean Sube
|
8dc1ea7f1f
|
declare missing cache
|
2023-02-13 18:15:02 -06:00 |
Sean Sube
|
f43f793148
|
fix type
|
2023-02-13 18:12:40 -06:00 |