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
|
5dfb09bd3c
|
fix(api): use existing progress counter for highres and loopback
|
2023-04-30 17:27:51 -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
|
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
|
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
|
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
|
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
|
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
|
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
|
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 |
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
|
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
|
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 |
Sean Sube
|
ac9b3d3b08
|
implement noise and blur source filters, move LPW in with other pipelines
|
2023-04-15 11:58:19 -05:00 |
Sean Sube
|
6854cbf23c
|
add attribution for ControlNet and Pix2Pix pipelines
|
2023-04-15 11:44:20 -05:00 |
Sean Sube
|
735f9d174f
|
apply lint
|
2023-04-15 11:20:11 -05:00 |
Sean Sube
|
c91916b4ca
|
hide img2img from pipelines list
|
2023-04-15 11:11:43 -05:00 |
Sean Sube
|
6effbea1eb
|
fix missing import
|
2023-04-15 10:57:37 -05:00 |
Sean Sube
|
7545d7c73d
|
add ControlNet UNet to existing checkpoints
|
2023-04-15 10:54:52 -05:00 |
Sean Sube
|
7f16b2f0ae
|
clean up CNet-only conversion code
|
2023-04-15 09:08:14 -05:00 |
Sean Sube
|
4fa9af7922
|
hide unused pipeline options
|
2023-04-15 09:02:17 -05:00 |
Sean Sube
|
0dd8272285
|
feat(api): convert CNet for existing diffusion models
|
2023-04-15 09:00:17 -05:00 |
Sean Sube
|
27954f3e65
|
feat(gui): add highres control to most tabs
|
2023-04-14 20:29:44 -05:00 |
Sean Sube
|
ad35c41c9d
|
feat(api): add highres to img2img mode for all pipelines
|
2023-04-14 08:54:21 -05:00 |
Sean Sube
|
c2f8ce5814
|
load UNet with ControlNet weights as needed
|
2023-04-13 23:05:00 -05:00 |
Sean Sube
|
06d55b0f1f
|
add filtered source to outputs
|
2023-04-13 22:51:59 -05:00 |
Sean Sube
|
f986b75c85
|
remove redundant HF download calls
|
2023-04-13 21:46:46 -05:00 |
Sean Sube
|
e2efa8c64e
|
fill in model names
|
2023-04-13 21:28:30 -05:00 |
Sean Sube
|
428fb06ecf
|
fix imports, add missing controlnet deps
|
2023-04-13 21:10:00 -05:00 |
Sean Sube
|
4df28a5ce7
|
feat: add menu for source image filters
|
2023-04-13 20:59:26 -05:00 |
Sean Sube
|
80d00e4477
|
feat(api): add source image filters for controlnet and others
|
2023-04-13 20:06:33 -05:00 |
Sean Sube
|
bd992398ae
|
lint, fix numpy error in pix2pix
|
2023-04-13 18:41:22 -05:00 |
Sean Sube
|
52d639ffed
|
use keyword args for networks
|
2023-04-13 00:02:47 -05:00 |
Sean Sube
|
982dc10e64
|
fix(api): make pix2pix work with scheduler with float sigma
|
2023-04-12 23:44:08 -05:00 |
Sean Sube
|
eb428e9541
|
fix strength
|
2023-04-12 23:33:01 -05:00 |
Sean Sube
|
8bf888959f
|
fix controlnet strength param
|
2023-04-12 23:30:59 -05:00 |
Sean Sube
|
c91bdb7679
|
fix serialization and scheduler
|
2023-04-12 23:26:16 -05:00 |
Sean Sube
|
6b51041ee8
|
feat(api): add list pipelines endpoint
|
2023-04-12 23:14:45 -05:00 |
Sean Sube
|
4aabf1ee27
|
wire up pipeline parameter, apply lint
|
2023-04-12 23:11:53 -05:00 |
Sean Sube
|
2af1530a7e
|
feat: make pipeline type a request parameter
|
2023-04-12 22:58:48 -05:00 |
Sean Sube
|
b2556809e9
|
apply lint
|
2023-04-12 20:06:13 -05:00 |
Sean Sube
|
555ec5a644
|
turn controlnet into a select list and localize its label, fix name lookup on server
|
2023-04-12 20:03:00 -05:00 |
Sean Sube
|
948bec6d0f
|
load controlnets from network models list
|
2023-04-12 08:51:16 -05:00 |
Sean Sube
|
9e017ee35d
|
feat: add parameter for ControlNet selection
|
2023-04-12 08:43:15 -05:00 |
Sean Sube
|
fbf576746c
|
attention fix for cnet export
|
2023-04-11 23:11:08 -05:00 |
Sean Sube
|
f49a4ddfdf
|
test controlnet pipe
|
2023-04-11 23:06:32 -05:00 |
Sean Sube
|
113f21fd34
|
remove safety checker
|
2023-04-11 19:40:08 -05:00 |
Sean Sube
|
a3daaf0112
|
initial integration of controlnet pipeline
|
2023-04-11 19:29:25 -05:00 |
Sean Sube
|
00be4f4721
|
fix(api): run PyTorch GC on ROCm devices (#323)
|
2023-04-11 08:34:21 -05:00 |
Sean Sube
|
de61e388a0
|
chore(api): remove some unused non-ONNX code paths
|
2023-04-11 08:26:21 -05:00 |
Sean Sube
|
1bfc7bee32
|
fix(api): run image workers as daemon procs
|
2023-04-10 23:23:08 -05:00 |
Sean Sube
|
76860f649a
|
feat(api): switch API server to Waitress (#159)
|
2023-04-10 23:21:13 -05:00 |
Sean Sube
|
e3540ac435
|
feat(api): add linear blend stage for chain pipeline (#183)
|
2023-04-10 21:15:57 -05:00 |
Sean Sube
|
0a6c99d57e
|
lint repeated factors
|
2023-04-10 21:04:48 -05:00 |
Sean Sube
|
2eaf85a43e
|
feat(api): add chain pipeline stages to load images from S3 or URL (#184, #185)
|
2023-04-10 21:03:19 -05:00 |
Sean Sube
|
7f6da8eeca
|
feat(api): add params for SwinIR large models
|
2023-04-10 19:13:49 -05:00 |
Sean Sube
|
6570d19fa3
|
feat(api): add SwinIR real x4 to base models
|
2023-04-10 19:04:52 -05:00 |
Sean Sube
|
2a7621c195
|
feat(api): add params for more SwinIR models
|
2023-04-10 19:02:12 -05:00 |
Sean Sube
|
e166989111
|
fix(api): use correct model for SwinIR
|
2023-04-10 17:57:42 -05:00 |
Sean Sube
|
62aa7e8473
|
feat(api): add initial support for BSRGAN and SwinIR upscaling (#153, #154)
|
2023-04-10 17:49:56 -05:00 |
Sean Sube
|
464bfd01b8
|
fix(api): apply FP16 patches to SD upscaling pipeline
|
2023-04-10 08:37:49 -05:00 |
Sean Sube
|
512f41135d
|
add logging when conv shapes do not match
|
2023-04-10 08:14:47 -05:00 |
Sean Sube
|
4db1737f98
|
apply lint
|
2023-04-10 08:09:29 -05:00 |
Sean Sube
|
7d2d865c19
|
convert LoHA T weights to same dtype, log shapes rather than data
|
2023-04-10 08:06:28 -05:00 |
Sean Sube
|
1ed51352c4
|
handle LoHA nodes with CP decomp
|
2023-04-10 08:06:22 -05:00 |
Sean Sube
|
88c7fd9f60
|
start handling Gemm nodes
|
2023-04-10 07:58:21 -05:00 |
Sean Sube
|
df02e20b34
|
only use mids once
|
2023-04-10 07:58:18 -05:00 |
Sean Sube
|
4947734add
|
flip axis
|
2023-04-10 07:58:15 -05:00 |
Sean Sube
|
87bba0e169
|
reverse factors
|
2023-04-10 07:58:12 -05:00 |
Sean Sube
|
d671cb8113
|
attempt layer by layer recomp
|
2023-04-10 07:58:08 -05:00 |
ForserX
|
7c4e729f85
|
Fix for LoHA weight and LoRA alpha read
|
2023-04-10 10:58:47 +03:00 |
Sean Sube
|
b6bb246032
|
apply lint
|
2023-04-09 20:34:10 -05:00 |
Sean Sube
|
9698e29268
|
lint(api): name context params consistently (#278)
|
2023-04-09 20:33:03 -05:00 |
Sean Sube
|
7f504a5b5b
|
feat: add parameter for highres iterations (#316)
|
2023-04-09 19:53:00 -05:00 |
Sean Sube
|
ff9ce03af5
|
fix(api): better handling for errors while converting checkpoint to torch (#165)
|
2023-04-09 18:05:44 -05:00 |
Sean Sube
|
34f1973707
|
feat(api): add support for PyTorch 2.0 (#292)
|
2023-04-09 16:07:40 -05:00 |
ForserX
|
d244ca9a9c
|
LoHA: Fixed blending error
|
2023-04-08 13:46:05 +03:00 |
Sean Sube
|
bd03edc2ad
|
fix(api): handle 3x3 kernels in LoHA networks
|
2023-04-07 20:54:42 -05:00 |
Sean Sube
|
7f7476bcbb
|
fix(api): use correct weight pairs in LoHA
|
2023-04-07 19:51:30 -05:00 |
Sean Sube
|
35432f1ab2
|
feat(api): experimental support for LoHA networks
|
2023-04-07 18:50:12 -05:00 |
Sean Sube
|
7e3ca8ab01
|
feat(api): support 3x3 kernels in LoRA and LoCONs
|
2023-04-06 23:34:55 -05:00 |
Sean Sube
|
25176fea4d
|
fix(api): correctly blend LoRAs into UNet during conversion
|
2023-04-06 23:34:06 -05:00 |
Sean Sube
|
1cfc538018
|
fix(api): ensure spiral grid coords are always whole pixels
|
2023-04-06 08:52:27 -05:00 |
Sean Sube
|
c8382dc689
|
feat(api): implement spiral tile order for non-square images
|
2023-04-04 20:02:13 -05:00 |
Sean Sube
|
0fdf4d3092
|
implement proper spiral grid coverage
|
2023-04-03 21:39:17 -05:00 |
Sean Sube
|
56ff9022da
|
fix(api): use min/max from config for more params
|
2023-04-01 18:14:24 -05:00 |
Sean Sube
|
85b332467e
|
fix(api): convert hidden states to fp32 before doing normalization on the CPU (#309)
|
2023-04-01 17:49:25 -05:00 |
Sean Sube
|
8e5971a293
|
fix lone else
|
2023-04-01 17:14:56 -05:00 |
Sean Sube
|
bbbef8d4ba
|
apply lint
|
2023-04-01 16:38:08 -05:00 |
Sean Sube
|
56c359c265
|
remove undefined names
|
2023-04-01 15:46:52 -05:00 |
Sean Sube
|
6bad599b3c
|
lint(api): move some chatty logs to trace level
|
2023-04-01 15:39:22 -05:00 |
Sean Sube
|
ed694aa64f
|
pass highres scale param to upscaling method
|
2023-04-01 15:15:56 -05:00 |
Sean Sube
|
e4f55af380
|
fix scale in upscale copy ctor
|
2023-04-01 14:59:59 -05:00 |
Sean Sube
|
89c3b2a403
|
correctly upscale highres tiles
|
2023-04-01 14:41:35 -05:00 |
Sean Sube
|
6d23491529
|
only run correction before highres when selected in options
|
2023-04-01 14:39:48 -05:00 |
Sean Sube
|
f451d8db74
|
feat: add method parameter for highres mode
|
2023-04-01 14:26:45 -05:00 |
Sean Sube
|
0f79f420b7
|
apply lint
|
2023-04-01 12:06:31 -05:00 |
Sean Sube
|
e0e09996de
|
fix(api): restart workers on HIP memory errors
|
2023-04-01 12:06:14 -05:00 |
Sean Sube
|
6aac0fe56b
|
fix(api): restart workers on MIOPEN memory errors
|
2023-04-01 11:59:03 -05:00 |
Sean Sube
|
3c2bab3156
|
fix highres steps max
|
2023-04-01 11:50:10 -05:00 |
Sean Sube
|
4f41145792
|
add highres to txt2img tab and request
|
2023-04-01 11:39:32 -05:00 |
Sean Sube
|
ba09748e94
|
feat(gui): add highres parameters
|
2023-04-01 11:26:10 -05:00 |
Sean Sube
|
f462d80cc4
|
run correction before highres
|
2023-03-31 23:55:51 -05:00 |
Sean Sube
|
ca80e92707
|
fix progress for highres, increase img2img steps
|
2023-03-31 23:52:33 -05:00 |
Sean Sube
|
4ab1b6cc88
|
resize tiles before running refinement steps
|
2023-03-31 23:39:25 -05:00 |
Sean Sube
|
66e938f9ab
|
split steps before and after highres
|
2023-03-31 23:32:05 -05:00 |
Sean Sube
|
cdaf1b8e5d
|
feat(api): add support for highres images
|
2023-03-31 23:27:16 -05:00 |
Sean Sube
|
93fcfd1422
|
fix(api): update LPW pipeline (#298)
|
2023-03-28 17:34:23 -05:00 |
Sean Sube
|
2bbc5d877f
|
lint(api): explicitly bind the device pool to shutdown callback
|
2023-03-27 17:13:45 -05:00 |
Sean Sube
|
c2f8fb1d31
|
fix(api): combine names for ONNX fp16 optimization
|
2023-03-27 08:55:01 -05:00 |
Sean Sube
|
73e9cf8b66
|
fix(api): disable internal fp16 for VAE encoder (#290)
|
2023-03-27 08:10:40 -05:00 |
Sean Sube
|
afa8f5e706
|
feat(api): add optimization for internal fp16 conversion
|
2023-03-26 22:41:12 -05:00 |
Sean Sube
|
0ea0442f02
|
apply lint
|
2023-03-26 18:24:22 -05:00 |
Sean Sube
|
4ddd69ba07
|
fix(api): watch for progress events from leaking workers
|
2023-03-26 18:22:16 -05:00 |
Sean Sube
|
e1219cca90
|
fix(api): close queues after stopping workers
|
2023-03-26 18:21:53 -05:00 |
Sean Sube
|
bb5d0635d1
|
sonar lint
|
2023-03-26 15:53:20 -05:00 |
Sean Sube
|
36bfccae09
|
fix(api): include worker totals in status endpoint
|
2023-03-26 15:49:43 -05:00 |
Sean Sube
|
2d2283e1eb
|
fix(api): attempt to read progress updates from recycled workers
|
2023-03-26 15:31:16 -05:00 |
Sean Sube
|
27500eccb5
|
fix(api): do not move jobs from pending to running until progress is reported
|
2023-03-26 14:22:59 -05:00 |
Sean Sube
|
0af406c47f
|
only enqueue jobs from progress worker
|
2023-03-26 14:02:57 -05:00 |
Sean Sube
|
ccf8d51e08
|
feat(api): split up status endpoint by job status
|
2023-03-26 11:57:00 -05:00 |
Sean Sube
|
ea36082e43
|
add job count to healthy worker logs
|
2023-03-26 11:53:06 -05:00 |
Sean Sube
|
8eab92a7df
|
define device on pending job
|
2023-03-26 11:49:58 -05:00 |
Sean Sube
|
83884bcafa
|
enqueue jobs on idle workers during progress check
|
2023-03-26 11:48:27 -05:00 |
Sean Sube
|
14ade83937
|
fix(api): enqueue next job when previous one finishes and after recycling worker
|
2023-03-26 11:41:45 -05:00 |
Sean Sube
|
f3ab25f671
|
lint(api): add start method to worker pool
|
2023-03-26 11:30:07 -05:00 |
Sean Sube
|
2b179bebac
|
fix(api): always reset job counter when creating new device worker
|
2023-03-26 11:22:03 -05:00 |
Sean Sube
|
55e44e8ac9
|
fix(api): increment job counter for worker when it starts a new job (#283)
|
2023-03-26 11:18:27 -05:00 |
Sean Sube
|
e552a5560f
|
feat(api): check device worker pool and recycle on a regular interval (#284)
|
2023-03-26 11:09:22 -05:00 |
Sean Sube
|
aeb71ad50a
|
lint lock name
|
2023-03-26 08:30:34 -05:00 |
Sean Sube
|
95a61f3573
|
fix(api): restart worker threads when their respective queues are full
|
2023-03-25 13:46:12 -05:00 |
Sean Sube
|
88f4713e23
|
fix(api): use lock when restarting workers
|
2023-03-25 09:47:51 -05:00 |
Sean Sube
|
2c47904057
|
lint(api): use constant for model filename
|
2023-03-24 08:14:19 -05:00 |
Sean Sube
|
6b4c046867
|
pass pool to threads
|
2023-03-22 22:58:46 -05:00 |
Sean Sube
|
86c1b29c31
|
lint(api): extract worker thread main functions (#279)
|
2023-03-22 22:55:34 -05:00 |
Sean Sube
|
4dd68ea6b6
|
fix(api): restart worker threads if they crash
|
2023-03-22 19:58:46 -05:00 |
Sean Sube
|
fa71d87e2c
|
apply lint
|
2023-03-21 22:19:50 -05:00 |
Sean Sube
|
e445d2afaa
|
fix hidden state conversion
|
2023-03-21 22:08:17 -05:00 |
Sean Sube
|
c8aad8554e
|
fix(api): convert back to model format after blending, convert samples as needed (#274)
|
2023-03-21 22:05:14 -05:00 |
Sean Sube
|
0315a8cbc6
|
fix(api): apply fp16 optimizations to LoRA and Textual Inversion blending
|
2023-03-21 21:45:27 -05:00 |
Sean Sube
|
4f6574c88e
|
sonar lint fixes
|
2023-03-21 17:11:38 -05:00 |
Sean Sube
|
a660d9f1b1
|
apply lint
|
2023-03-20 22:28:48 -05:00 |
Sean Sube
|
7f459e0e29
|
correct extensions in ignore list
|
2023-03-19 23:31:11 -05:00 |
Sean Sube
|
f779fbd5cc
|
fix(api): remove prefix before downloading for HF hub
|
2023-03-19 23:30:31 -05:00 |
Sean Sube
|
19712262e6
|
fix(api): filter out temporary files from model lists (#271)
|
2023-03-19 23:26:05 -05:00 |
Sean Sube
|
ae3bcf3b8b
|
fix(api): add HF hub download to fetch logic for Inversion concepts
|
2023-03-19 20:32:21 -05:00 |
Sean Sube
|
0732058aa8
|
feat(api): detect Textual Inversion type from keys (#262)
|
2023-03-19 20:16:52 -05:00 |
Sean Sube
|
fe498b16f0
|
fix(api): embed Inversion concepts using their name
|
2023-03-19 19:01:22 -05:00 |
Sean Sube
|
19d4d554c3
|
fix(api): add base token for Textual Inversion as sum layer
|
2023-03-19 18:04:21 -05:00 |
Sean Sube
|
aefa5b4613
|
fix(api): clear job cancelled flag when worker starts a new job (#269)
|
2023-03-19 17:57:14 -05:00 |
Sean Sube
|
8acc15f52e
|
fix(api): better error handling around tensor loading
|
2023-03-19 15:38:43 -05:00 |
Sean Sube
|
07622690dc
|
fix(api): use Torch JIT and fallback loading
|
2023-03-19 15:31:35 -05:00 |
Sean Sube
|
b797b3d616
|
fix(api): do not assume file extension for additional networks
|
2023-03-19 15:27:51 -05:00 |
Sean Sube
|
33008531e9
|
fix(api): load blending tensors onto CPU
|
2023-03-19 15:13:54 -05:00 |
Sean Sube
|
1c631c28d3
|
feat(api): add flag for ORT float16 optimizations
|
2023-03-19 11:59:35 -05:00 |
Sean Sube
|
b8f816323e
|
fix(api): correctly load server env for conversion
|
2023-03-19 10:39:09 -05:00 |
Sean Sube
|
98e488319c
|
use fp16 optimization flag during conversion, add to admin docs
|
2023-03-19 10:33:46 -05:00 |
Sean Sube
|
edd32f6044
|
apply lint, fix up log levels
|
2023-03-19 10:25:09 -05:00 |
Sean Sube
|
6f283c5c02
|
proxy nets that need fp16 conversion
|
2023-03-19 10:18:22 -05:00 |
Sean Sube
|
bbd4c0fd72
|
dtype logging for potential fp16 params
|
2023-03-19 09:59:08 -05:00 |
Sean Sube
|
243a2d9df6
|
apply lint
|
2023-03-19 09:29:06 -05:00 |
Sean Sube
|
27bd10d2f7
|
fix(api): restrict CLIP skip token to integers, allow negative LoRA and Inversion weights
|
2023-03-19 09:13:40 -05:00 |
Sean Sube
|
e6d7d30225
|
move tensor logs to trace level
|
2023-03-19 09:11:55 -05:00 |
Sean Sube
|
0e43acc0f7
|
fix(api): fetch sources and additional networks by default
|
2023-03-19 09:09:03 -05:00 |
Sean Sube
|
b82246fdab
|
parse CLIP skip count from prompt
|
2023-03-19 08:43:39 -05:00 |
Sean Sube
|
46d1b5636d
|
normalize hidden states without using CLIP model class
|
2023-03-19 08:40:06 -05:00 |
Sean Sube
|
2ef00599b6
|
experimental CLIP skip
|
2023-03-19 08:17:40 -05:00 |
Sean Sube
|
9a949806d3
|
apply lint
|
2023-03-18 21:48:51 -05:00 |
Sean Sube
|
645da3d1a6
|
fix(api): allow dash in network names
|
2023-03-18 21:48:20 -05:00 |
Sean Sube
|
a9e55ff9f2
|
remove references to inversion parameter
|
2023-03-18 19:47:22 -05:00 |
Sean Sube
|
5363037ab1
|
use network name in label key
|
2023-03-18 19:43:00 -05:00 |
Sean Sube
|
61d12b8ebd
|
include model type in label when present
|
2023-03-18 19:34:03 -05:00 |
Sean Sube
|
f8f36e602c
|
feat(api): load i18n labels from additional networks
|
2023-03-18 19:25:22 -05:00 |
Sean Sube
|
2e89fd43d3
|
fix(api): only remove running jobs from running state
|
2023-03-18 19:21:40 -05:00 |
Sean Sube
|
716e009611
|
serialize networks
|
2023-03-18 19:21:15 -05:00 |
Sean Sube
|
2d112104fb
|
feat: show additional networks in client
|
2023-03-18 19:15:23 -05:00 |
Sean Sube
|
e5862d178c
|
fix(api): assume inversion tokens are embeddings for now
|
2023-03-18 18:35:11 -05:00 |
Sean Sube
|
ed8a7c8934
|
fix(api): move prompt parsing logs to trace level
|
2023-03-18 17:32:32 -05:00 |
Sean Sube
|
d321321507
|
fix fields
|
2023-03-18 17:28:59 -05:00 |
Sean Sube
|
1d52dc684d
|
init last progress on worker context
|
2023-03-18 17:27:41 -05:00 |
Sean Sube
|
e08a9aa2ab
|
add pending job list to pool
|
2023-03-18 17:26:28 -05:00 |
Sean Sube
|
8cbdad3a71
|
feat(api): add pending field to image ready response
|
2023-03-18 17:25:13 -05:00 |
Sean Sube
|
15b6e036e1
|
fix(api): maintain list of pending jobs
|
2023-03-18 17:15:18 -05:00 |
Sean Sube
|
588c8c7fdb
|
fix(api): track last progress within worker
|
2023-03-18 15:32:49 -05:00 |
Sean Sube
|
5106dd48a9
|
remove another ref to finished queue
|
2023-03-18 15:27:07 -05:00 |
Sean Sube
|
b026566ccb
|
remove remaining references to finished queue and worker
|
2023-03-18 15:26:19 -05:00 |
Sean Sube
|
d1565b056e
|
apply lint, make missing images an error
|
2023-03-18 15:16:41 -05:00 |
Sean Sube
|
7cf5554bef
|
feat(api): add error flag to image ready response
|
2023-03-18 15:13:42 -05:00 |
Sean Sube
|
f336cc5d02
|
fix(api): provide noise parameter to SD upscaling (#196)
|
2023-03-18 14:02:30 -05:00 |
Sean Sube
|
a9456f4a16
|
fix(api): bump default model cache to 5 (#242)
|
2023-03-18 13:47:39 -05:00 |
Sean Sube
|
aec540a524
|
feat(api): add server setting for CUDA memory limit (#211)
|
2023-03-18 13:40:37 -05:00 |
Sean Sube
|
af326a784f
|
fix(api): avoid loading encoder twice when using LoRAs and inversions together
|
2023-03-18 13:39:35 -05:00 |
Sean Sube
|
9f9b73b780
|
fix(api): use ORT session for correct device when loading blended nets
|
2023-03-18 13:39:04 -05:00 |
Sean Sube
|
84718e5928
|
feat(scripts): add env debug script (#191)
|
2023-03-18 13:16:59 -05:00 |
Sean Sube
|
5b308d5fc1
|
lint(api): improve logging for inversion tokens
|
2023-03-18 12:04:39 -05:00 |
Sean Sube
|
2cb0a6be3c
|
more lint
|
2023-03-18 11:55:06 -05:00 |
Sean Sube
|
e104c81e19
|
apply lint
|
2023-03-18 11:50:09 -05:00 |
Sean Sube
|
a1f17c0877
|
fix dest paths for blended nets
|
2023-03-18 11:37:16 -05:00 |
Sean Sube
|
f465120cad
|
use conversion dest path when applying additional nets
|
2023-03-18 11:34:05 -05:00 |
Sean Sube
|
1f6105a8fe
|
make blend functions take tuples rather than split lists
|
2023-03-18 10:50:48 -05:00 |
Sean Sube
|
40c92f054b
|
add networks as conversion arg
|
2023-03-18 10:22:24 -05:00 |
Sean Sube
|
5dd14c19ce
|
feat(api): build a single token for multi-layer Textual Inversions (#212)
|
2023-03-18 07:47:36 -05:00 |
Sean Sube
|
af62c1c3b6
|
apply lint
|
2023-03-18 07:41:29 -05:00 |
Sean Sube
|
e3bf04ab8f
|
feat(api): add section to extras file for additional networks
|
2023-03-18 07:40:57 -05:00 |
Sean Sube
|
c3979246df
|
make blending happen once after conversion
|
2023-03-18 07:14:22 -05:00 |
Sean Sube
|
32b2a76a0b
|
fix(api): download additional networks to their own subdir in models
|
2023-03-18 07:07:05 -05:00 |
Sean Sube
|
84bd852837
|
feat(api): blend LoRAs and Textual Inversions from extras file
|
2023-03-18 07:01:24 -05:00 |
Sean Sube
|
1d44f985a4
|
fix(api): restore python 3.8 compatibility
|
2023-03-17 23:07:10 -05:00 |
Sean Sube
|
c943a55a44
|
fix(api): move imports of patches modules into function
|
2023-03-17 23:06:39 -05:00 |
Sean Sube
|
226710a015
|
fix(api): use exception level logs
|
2023-03-16 22:29:07 -05:00 |
Sean Sube
|
d3a17ec648
|
fix logging metadata
|
2023-03-16 21:45:57 -05:00 |
Sean Sube
|
4b832f3d8d
|
more lint, more trace
|
2023-03-16 20:22:20 -05:00 |
Sean Sube
|
9f0a6f134e
|
fix(api): move model conversion messages to trace level
|
2023-03-16 19:37:25 -05:00 |
Sean Sube
|
c397c1e42a
|
apply sonar lint
|
2023-03-16 18:47:04 -05:00 |
Sean Sube
|
f4c0c52029
|
apply lint
|
2023-03-16 18:35:30 -05:00 |
Sean Sube
|
959798eabb
|
feat(api): add trace log level
|
2023-03-16 18:34:47 -05:00 |
Sean Sube
|
c8c5e9f42e
|
fix(api): handle more out-of-memory errors in the workers
|
2023-03-16 18:34:28 -05:00 |
Sean Sube
|
315e5a3837
|
fix(api): add theoretical support for 3x3 conv in LoRA
|
2023-03-15 19:37:17 -05:00 |
Sean Sube
|
8e8e230ffd
|
apply lint
|
2023-03-15 19:27:29 -05:00 |
Sean Sube
|
506cf9f65f
|
feat(api): blend Textual Inversions from prompt
|
2023-03-15 17:14:52 -05:00 |
Sean Sube
|
973ad0f682
|
fix inversion var
|
2023-03-15 08:57:17 -05:00 |
Sean Sube
|
b451196dfa
|
add prefix to inversion path
|
2023-03-15 08:55:35 -05:00 |
Sean Sube
|
421ed632b1
|
only use name from first inversion
|
2023-03-15 08:53:15 -05:00 |
Sean Sube
|
b2eb406197
|
fix(api): handle CUDA memory errors in workers
|
2023-03-15 08:51:29 -05:00 |
Sean Sube
|
a2e21f427f
|
feat(api): load Textual Inversions from prompt
|
2023-03-15 08:51:12 -05:00 |
Sean Sube
|
829cedc934
|
fix error when prompt has no LoRA tokens
|
2023-03-15 08:35:44 -05:00 |
Sean Sube
|
a7f77a033d
|
feat(api): parse LoRA weights from prompt
|
2023-03-15 08:30:31 -05:00 |
Sean Sube
|
45166f281e
|
apply lint
|
2023-03-14 23:32:47 -05:00 |
Sean Sube
|
91210ee236
|
remove extra param, correct output path
|
2023-03-14 23:32:18 -05:00 |
Sean Sube
|
143904fc51
|
feat(api): parse LoRA names from prompt
|
2023-03-14 22:28:18 -05:00 |
Sean Sube
|
03f4e1b922
|
start wiring LoRAs into prompt
|
2023-03-14 22:10:33 -05:00 |
Sean Sube
|
ce05e76947
|
load base models from model dir, use correct ORT provider
|
2023-03-14 21:57:37 -05:00 |
Sean Sube
|
56a4519818
|
test loading UNet and encoder with LoRAs
|
2023-03-14 21:27:23 -05:00 |
Sean Sube
|
2a8c85cb3e
|
feat(api): blend ONNX models in memory
|
2023-03-14 19:38:27 -05:00 |
Sean Sube
|
4c17edb267
|
feat(api): add conversion script for LoRAs from sd-scripts (#213)
|
2023-03-14 18:17:00 -05:00 |
Sean Sube
|
0b1aa26be5
|
blend LoRAs into a valid ONNX UNet (#213)
|
2023-03-12 13:38:51 -05:00 |
Sean Sube
|
919f88c7d0
|
fix(tests): clear cache between tests
|
2023-03-11 13:59:09 -06:00 |
Sean Sube
|
9555a7a3ea
|
lint(api): only log new worker message if some workers need to be restarted
|
2023-03-11 13:30:54 -06:00 |
Sean Sube
|
9c65922526
|
lint(api): remove unnecessary delete calls
|
2023-03-11 13:30:33 -06:00 |
Sean Sube
|
575cb8831b
|
fix(api): make cache global within each worker process (#227)
|
2023-03-11 13:30:11 -06:00 |
Sean Sube
|
01d3519aa3
|
fix(api): remove unused num_workers logic from server
|
2023-03-11 08:17:27 -06:00 |
Sean Sube
|
8e5d0871c8
|
log which cache items have been pruned
|
2023-03-11 08:06:22 -06:00 |
Sean Sube
|
cb2b054fde
|
fix(api): only run GC when a diffusion model has been replaced
|
2023-03-11 08:06:03 -06:00 |
Sean Sube
|
cd11e8f3ab
|
use a proper fake config class
|
2023-03-10 18:57:01 -06:00 |
Sean Sube
|
df1504d9b0
|
use fake VAE config
|
2023-03-10 18:50:01 -06:00 |
Sean Sube
|
555de5e07d
|
fix(api): correct VAE config check
|
2023-03-10 18:46:07 -06:00 |
Sean Sube
|
e960ee1d85
|
fix(api): patch VAE for SD upscaling on older versions of diffusers (#234)
|
2023-03-10 18:42:11 -06:00 |
Sean Sube
|
6bdaa4a4ed
|
fix(api): make stage-specific source image optional in GFPGAN stage (#233)
|
2023-03-10 18:15:31 -06:00 |
Sean Sube
|
da3d95ff49
|
fix(api): include bottom margin when calculating output size
|
2023-03-10 17:59:16 -06:00 |
Sean Sube
|
fe437e8cd4
|
feat: add support for UniPC multistep scheduler (#226)
|
2023-03-09 08:00:59 -06:00 |
Sean Sube
|
21c60709bc
|
fix(api): move prompt splitting logging to debug level
|
2023-03-08 22:55:58 -06:00 |
Sean Sube
|
25c41c8d66
|
feat(api): add an option to remove temporary Torch files after converting to ONNX (#122)
|
2023-03-08 21:38:17 -06:00 |
Sean Sube
|
6e71775782
|
fix(api): load LPW pipeline from local source (#224)
|
2023-03-08 19:47:22 -06:00 |
Sean Sube
|
7bd98d4a46
|
apply lint
|
2023-03-07 23:57:39 -06:00 |
Sean Sube
|
bc21165fd9
|
fix path ref
|
2023-03-07 23:54:32 -06:00 |
Sean Sube
|
c0b88733b8
|
fix(api): check for cache rather than converted model before fetch
|
2023-03-07 23:51:27 -06:00 |
Sean Sube
|
843e2f1ff3
|
feat(api): look for an index file when checking for converted models (#222)
|
2023-03-07 23:40:04 -06:00 |
Sean Sube
|
30b08c6d6d
|
fix(api): log correct token when converting Textual Inversions
|
2023-03-07 23:05:53 -06:00 |
Sean Sube
|
6542646005
|
fix(api): match format in schema for sd-concepts
|
2023-03-07 22:53:05 -06:00 |
Sean Sube
|
85ff700525
|
apply lint
|
2023-03-07 22:40:17 -06:00 |
Sean Sube
|
0a4f83ac0f
|
feat(api): add range syntax to expand numbered tokens (#179)
|
2023-03-07 20:48:26 -06:00 |
Sean Sube
|
66c42485cb
|
feat(api): add support for extremely long prompts
|
2023-03-07 19:00:25 -06:00 |
Sean Sube
|
c45915e558
|
fix(api): use server model path while converting SD checkpoints (#221)
|
2023-03-07 18:55:14 -06:00 |
Sean Sube
|
9d9bd1a639
|
apply lint
|
2023-03-07 08:02:53 -06:00 |
Sean Sube
|
af1c3c7839
|
fix(api): track items removed from cache
|
2023-03-06 07:34:09 -06:00 |
Sean Sube
|
339868de4d
|
fix(api): use pipeline class name in cache key
|
2023-03-05 21:54:09 -06:00 |
Sean Sube
|
55576ae87c
|
fix(api): include cache keys in logs
|
2023-03-05 21:50:01 -06:00 |
Sean Sube
|
35dc8a0bc4
|
improve exit logging
|
2023-03-05 21:37:39 -06:00 |
Sean Sube
|
c0a01efef4
|
fix(api): track currently active worker for each device
|
2023-03-05 21:28:21 -06:00 |
Sean Sube
|
57fed94337
|
fix(api): exit worker on memory allocation errors
|
2023-03-05 21:11:33 -06:00 |
Sean Sube
|
cb460a0c59
|
fix(api): add worker PID to log messages
|
2023-03-05 20:25:02 -06:00 |
Sean Sube
|
4ae4ce176c
|
fix(api): attempt to recycle leaking workers when a job finishes
|
2023-03-05 20:13:28 -06:00 |
Sean Sube
|
3a4928e59b
|
fix(api): prevent workers from blocking on their progress queues
|
2023-03-05 20:07:06 -06:00 |
Sean Sube
|
edc55ae8b4
|
fix(api): finished job notification should not block worker
|
2023-03-05 19:53:44 -06:00 |
Sean Sube
|
cfc20d3133
|
fix(api): improve cache logging
|
2023-03-05 19:30:52 -06:00 |
Sean Sube
|
39b9741b24
|
fix(api): show VRAM percent in logs
|
2023-03-05 19:23:23 -06:00 |
Sean Sube
|
7a3a81a4ef
|
fix(api): track and repeatedly attempt to recycle leaking workers (#219)
|
2023-03-05 18:58:13 -06:00 |
Sean Sube
|
468870e2e7
|
apply lint
|
2023-03-05 08:14:47 -06:00 |
Sean Sube
|
00fa9c5807
|
feat(api): collect labels from textual inversions (#144)
|
2023-03-05 08:00:25 -06:00 |
Sean Sube
|
6d2dd0a043
|
apply lint
|
2023-03-05 07:20:43 -06:00 |
Sean Sube
|
539d0febce
|
load extras during server init
|
2023-03-05 07:20:37 -06:00 |
Sean Sube
|
537c67619e
|
fix strings endpoint
|
2023-03-05 07:20:34 -06:00 |
Sean Sube
|
5d459ab17c
|
feat: allow users to add their own labels for models (#144)
|
2023-03-05 07:20:31 -06:00 |
Sean Sube
|
628812fb0b
|
fix(api): correct conversion imports
|
2023-03-04 23:03:15 -06:00 |
Sean Sube
|
43d4d4b86d
|
chore(api): dedupe model listing logic
|
2023-03-04 22:26:27 -06:00 |
Sean Sube
|
10fbafaff0
|
fix(api): correct imports
|
2023-03-04 22:25:49 -06:00 |
Sean Sube
|
7b88371832
|
lint(api): use load pattern in server submodule
|
2023-03-04 22:13:09 -06:00 |
Sean Sube
|
bb1d945761
|
lint(api): submodules for major libs
|
2023-03-04 22:12:35 -06:00 |
Sean Sube
|
39d36618e6
|
feat(api): add support for custom tokens for textual inversions (#179)
|
2023-03-02 23:32:20 -06:00 |
Sean Sube
|
22e9d54408
|
apply lint
|
2023-03-02 22:32:37 -06:00 |
Sean Sube
|
be1c8cd79f
|
feat(api): support ckpt inversions by splitting multivector embeddings into multiple tokens
|
2023-03-02 22:11:27 -06:00 |
Sean Sube
|
9a0d2051fb
|
fix(api): better logging when converting textual inversions
|
2023-03-02 07:57:59 -06:00 |
Sean Sube
|
46aac263d5
|
feat(api): initial support for textual inversion embeddings from civitai and others (#179)
|
2023-03-01 19:09:51 -06:00 |
Sean Sube
|
1f3a5f6f3c
|
fix(api): track completed jobs for each device worker (#170)
|
2023-03-01 19:09:18 -06:00 |
Sean Sube
|
4b77a00ca7
|
fix(api): load tokenizer with textual inversions
|
2023-03-01 19:08:31 -06:00 |
Sean Sube
|
21fc7c5968
|
fix(api): mark all convert methods as no_grad
|
2023-03-01 08:26:40 -06:00 |
Sean Sube
|
3f9f94fcb5
|
apply lint, remove unused
|
2023-02-28 23:05:17 -06:00 |
Sean Sube
|
2f4ab20f61
|
use filename for tensors
|
2023-02-28 22:49:53 -06:00 |
Sean Sube
|
74aae1b027
|
fix(api): write external weights into same directory as optimized model
|
2023-02-28 22:47:02 -06:00 |
Sean Sube
|
dbf9eaf1a4
|
fix(api): run shape inference before converting models to fp16
per discussion in https://github.com/microsoft/onnxruntime/issues/14827
|
2023-02-28 22:36:45 -06:00 |
Sean Sube
|
86984be5d4
|
better error logging during conversion
|
2023-02-28 22:36:36 -06:00 |
Sean Sube
|
9ef89db8b0
|
extract tensors after conversion
|
2023-02-28 22:36:33 -06:00 |
Sean Sube
|
7e65e21410
|
reload model from proto file before converting
|
2023-02-28 22:36:26 -06:00 |
Sean Sube
|
2210ee849b
|
only convert inner nodes with ORT conversion helpers
|
2023-02-28 22:26:04 -06:00 |
Sean Sube
|
b532570f54
|
fix(api): reuse ORT session params for each device
|
2023-02-28 22:24:38 -06:00 |
Sean Sube
|
565873b3ae
|
test half precision on all diffusion models
|
2023-02-28 22:23:55 -06:00 |
Sean Sube
|
1f9efb433a
|
apply lint
|
2023-02-28 22:04:55 -06:00 |
Sean Sube
|
12fb7f52bb
|
fix(api): sanitize filenames in user input
|
2023-02-28 21:56:12 -06:00 |
Sean Sube
|
c99aa67220
|
name threads, max queues, type/lint fixes
|
2023-02-28 21:44:52 -06:00 |
Sean Sube
|
c95ac1fbdd
|
avoid terminating workers because it breaks their queues
|
2023-02-28 08:53:17 -06:00 |
Sean Sube
|
0011f079d4
|
daemonize queue collectors
|
2023-02-28 06:55:15 -06:00 |
Sean Sube
|
cad0d37604
|
some pending queue logging
|
2023-02-27 23:43:38 -06:00 |
Sean Sube
|
4ae3d9caa2
|
remove task done
|
2023-02-27 23:18:37 -06:00 |
Sean Sube
|
7e0ccdb1af
|
remove pending queues after joining
|
2023-02-27 23:14:20 -06:00 |
Sean Sube
|
1ce98ace33
|
add value error handling
|
2023-02-27 23:12:53 -06:00 |
Sean Sube
|
f7f438e767
|
directly rejoin pool
|
2023-02-27 23:03:42 -06:00 |
Sean Sube
|
da6ae5d62f
|
more logging around shutdown, close queues
|
2023-02-27 23:01:26 -06:00 |
Sean Sube
|
988088d64e
|
quit workers on keyboard signal
|
2023-02-27 22:52:43 -06:00 |
Sean Sube
|
953e5abd36
|
handle empty errors
|
2023-02-27 22:45:29 -06:00 |
Sean Sube
|
136759285d
|
set queue timeouts
|
2023-02-27 22:37:43 -06:00 |
Sean Sube
|
0793b61c3a
|
consistently pass job key to workers
|
2023-02-27 22:25:53 -06:00 |
Sean Sube
|
06f06f5a11
|
error handling in all threads
|
2023-02-27 19:48:51 -06:00 |
Sean Sube
|
113ad05293
|
typo
|
2023-02-27 17:36:26 -06:00 |
Sean Sube
|
2327b24022
|
join all threads
|
2023-02-27 17:35:31 -06:00 |
Sean Sube
|
66a20e60fe
|
run logger in a thread, clean up status
|
2023-02-27 17:14:53 -06:00 |
Sean Sube
|
13395933dc
|
always put progress in active jobs
|
2023-02-26 20:41:16 -06:00 |
Sean Sube
|
a37d1a4550
|
use progress queue
|
2023-02-26 20:37:22 -06:00 |
Sean Sube
|
401ee20526
|
fix finished flag
|
2023-02-26 20:13:16 -06:00 |
Sean Sube
|
525ee24e91
|
track started and finished jobs
|
2023-02-26 20:09:42 -06:00 |
Sean Sube
|
eb82e73e59
|
initialize list of finished jobs
|
2023-02-26 15:26:54 -06:00 |
Sean Sube
|
b931da1d2c
|
fix imports, lint
|
2023-02-26 15:21:58 -06:00 |
Sean Sube
|
85118d17c6
|
clear worker flags between jobs, attempt to record finished jobs again
|
2023-02-26 15:06:40 -06:00 |
Sean Sube
|
d1961afdbc
|
re-implement cancellation
|
2023-02-26 14:36:32 -06:00 |
Sean Sube
|
584dddb5d6
|
lint all the new stuff
|
2023-02-26 14:15:30 -06:00 |
Sean Sube
|
b880b7a121
|
set process titles, terminate workers
|
2023-02-26 13:09:24 -06:00 |
Sean Sube
|
6502e1e3c8
|
recycle worker pool after 10 jobs
|
2023-02-26 12:58:38 -06:00 |
Sean Sube
|
e0737e9e08
|
update progress and finished flag from worker
|
2023-02-26 12:51:11 -06:00 |
Sean Sube
|
f115326da7
|
apply patches within workers
|
2023-02-26 12:32:48 -06:00 |
Sean Sube
|
e1d0ad54b7
|
lock per worker, torch before ORT
|
2023-02-26 12:24:51 -06:00 |
Sean Sube
|
d765a6f01b
|
make logger start up well
|
2023-02-26 11:16:33 -06:00 |
Sean Sube
|
6998e8735c
|
rejoin worker pool
|
2023-02-26 10:47:31 -06:00 |
Sean Sube
|
06c74a7a96
|
feat(api): remove Flask app from global scope
|
2023-02-26 10:15:12 -06:00 |
Sean Sube
|
943281feb5
|
wire up worker jobs
|
2023-02-25 23:55:30 -06:00 |
Sean Sube
|
f898de8c54
|
background workers, logger
|
2023-02-25 23:49:39 -06:00 |
Sean Sube
|
e46a1e5fd0
|
begin switching to per-device torch mp workers
|
2023-02-25 23:16:32 -06:00 |
Sean Sube
|
e03b637f54
|
fix(api): only run CUDA GC on CUDA devices
|
2023-02-25 22:32:01 -06:00 |
Sean Sube
|
cb8e9e7080
|
fix scheduler caching
|
2023-02-25 13:14:34 -06:00 |
Sean Sube
|
18f59f034d
|
avoid passing None as encoder
|
2023-02-25 13:12:58 -06:00 |
Sean Sube
|
3626d69f40
|
fix(api): correctly handle missing inversion param
|
2023-02-25 12:54:51 -06:00 |
Sean Sube
|
973b0a9352
|
add context arg
|
2023-02-25 12:13:22 -06:00 |
Sean Sube
|
d88f13cbd7
|
lint LoRA code and extras, replace public paths with context ones
|
2023-02-25 12:12:10 -06:00 |
Sean Sube
|
ce74183e97
|
blend LoRAs into existing model
|
2023-02-25 11:53:43 -06:00 |
Sean Sube
|
e8b5ff250d
|
add none option to inversion menu
|
2023-02-25 11:53:37 -06:00 |
Sean Sube
|
7ad8385c5b
|
load from encoder path, add default param
|
2023-02-25 11:53:34 -06:00 |
Sean Sube
|
2e7de16778
|
feat: add a way to select textual inversions
|
2023-02-25 11:53:32 -06:00 |
Sean Sube
|
45f5fca383
|
use correct fn to make dirs
|
2023-02-25 11:53:29 -06:00 |
Sean Sube
|
3dfaef041e
|
fix devices, make subdir
|
2023-02-25 11:53:26 -06:00 |
Sean Sube
|
3f4b3fa322
|
load CLIP on training device
|
2023-02-25 11:53:23 -06:00 |
Sean Sube
|
6b4ced2608
|
load inversions from extras file
|
2023-02-25 11:53:20 -06:00 |
Sean Sube
|
9dedfc7b28
|
add experimental LoRA blender
|
2023-02-25 11:53:17 -06:00 |
Sean Sube
|
a31f7b9e1f
|
feat(api): convert Textual Inversion weights
|
2023-02-25 11:53:13 -06:00 |
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 |
Sean Sube
|
0709c1dbf0
|
feat(api): add GFPGAN and Real ESRGAN to model cache
|
2023-02-13 18:10:11 -06:00 |
Sean Sube
|
e9472bc005
|
feat(api): add model cache for diffusion models
|
2023-02-13 18:04:46 -06:00 |
Sean Sube
|
7fa1783be4
|
feat: implement blend tab and copy buttons (#62)
|
2023-02-13 17:34:42 -06:00 |
Sean Sube
|
d6201c9d32
|
feat(api): add an option for custom checkpoint config to extras file (fixes #130)
|
2023-02-12 14:10:30 -06:00 |
Sean Sube
|
82487f5771
|
fix(api): separate Real ESRGAN upsampler from GFPGAN stage (#87)
|
2023-02-12 13:50:28 -06:00 |
Sean Sube
|
ae8aa2ea7a
|
apply lint
|
2023-02-12 13:32:50 -06:00 |
Sean Sube
|
034be3266e
|
fix(api): accumulate progress from inpaint pipelines (#90)
|
2023-02-12 13:16:17 -06:00 |
Sean Sube
|
aaf82a42f1
|
fix(api): use correct var for diffusion scheduler cache
|
2023-02-12 12:47:55 -06:00 |
Sean Sube
|
feb7e7d57b
|
pass progress to txt2img and img2img
|
2023-02-12 12:38:06 -06:00 |
Sean Sube
|
2f6a3afddb
|
pass progress on to most stages
|
2023-02-12 12:33:36 -06:00 |
Sean Sube
|
9d1c5dca52
|
fix(api): correctly handle partial extras files
|
2023-02-12 12:23:03 -06:00 |
Sean Sube
|
06d4f32574
|
make progress callbacks optional
|
2023-02-12 12:22:11 -06:00 |
Sean Sube
|
d9fc908592
|
feat(api): collect progress from chain pipelines (#90)
|
2023-02-12 12:17:36 -06:00 |
Sean Sube
|
27a3fa8f51
|
feat: show model in image card, use labels for model and scheduler (#104)
|
2023-02-12 09:51:35 -06:00 |
Sean Sube
|
9c5043e9d0
|
fix(api): correctly cache diffusers scheduler
|
2023-02-12 09:33:13 -06:00 |
Sean Sube
|
1179092028
|
feat(api): intercept model downloads in libs and use cached copy (fixes #95)
|
2023-02-12 09:28:37 -06:00 |
Sean Sube
|
8ea33e9874
|
fix(api): patch various download fns to use cache (#95)
|
2023-02-12 06:25:44 -06:00 |
Sean Sube
|
1f06b502d5
|
fix(api): always fill inpaint/outpaint mask with white
|
2023-02-11 18:55:22 -06:00 |
Sean Sube
|
ffbe0929e6
|
apply lint
|
2023-02-11 18:10:36 -06:00 |
Sean Sube
|
3a290822eb
|
feat(api): pass tile order to inpaint and outpaint pipelines
|
2023-02-11 18:00:18 -06:00 |