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 |
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
|
9a2f35e181
|
fix(api): use Waitress for Windows bundle
|
2023-04-10 23:22:32 -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
|
23fb752bb6
|
fix(api): correct type of model field in extras
|
2023-04-10 17:59:32 -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
|
b4095cc2db
|
fix(scripts): show data shape in initializer diff
|
2023-04-10 08:26:24 -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
|
fea9185707
|
fix(api): open web browser when launching Windows bundle
|
2023-04-09 20:06:43 -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
|
89ebbb8536
|
feat(api): update requirements for torch 2.0
|
2023-04-09 17:19:40 -05:00 |
Sean Sube
|
34f1973707
|
feat(api): add support for PyTorch 2.0 (#292)
|
2023-04-09 16:07:40 -05:00 |
Sean Sube
|
a291dc8980
|
fix(api): add conversion to Windows bundle ep
|
2023-04-09 15:01:38 -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
|
83dbd479b9
|
update params version for highres
|
2023-04-01 17:15:09 -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
|
4e481e4864
|
add highres to server params
|
2023-04-01 11:31:34 -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
|
27c6e71890
|
fix(api): add coloredlogs to base deps
|
2023-03-30 23:43:40 -05:00 |
Sean Sube
|
205ff3e403
|
feat(exe): add specs and launch scripts for Windows EXE bundle (#305)
|
2023-03-30 21:02:23 -05:00 |
Sean Sube
|
1cd436d643
|
feat(api): add setup scripts for Windows
|
2023-03-30 21:01:55 -05:00 |
Sean Sube
|
b4ac7c6259
|
chore(release): 0.9.0
|
2023-03-28 18:57:24 -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
|
d7e548037a
|
fix(tests): make release tests fail if image was not successful (#287)
|
2023-03-26 20:45:27 -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
|
dca8a97796
|
feat(api): pin pytorch versions and update nightly ORT
|
2023-03-26 18:21:21 -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
|
580d621209
|
fix(api): make format list in schema match code
|
2023-03-26 08:28:14 -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
|
391a707c84
|
fix(scripts): check Torch CUDA devices (#242)
|
2023-03-18 13:54:12 -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
|
c465b61fb5
|
fix(scripts): check for ORT modules
|
2023-03-18 13:21:17 -05:00 |
Sean Sube
|
84718e5928
|
feat(scripts): add env debug script (#191)
|
2023-03-18 13:16:59 -05:00 |
Sean Sube
|
17e4fd7b06
|
feat(scripts): add model type guessing script (#210)
|
2023-03-18 13:01:05 -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
|
6cd0b4f7eb
|
fix extras schema
|
2023-03-18 10:24:16 -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
|
fc9245f4ee
|
fix(buil): limit torchvision versions to match torch
|
2023-03-17 22:42:37 -05:00 |
Sean Sube
|
a21750ff8e
|
fix(build): pin Torch to 1.x for now
|
2023-03-17 21:31:15 -05:00 |
Sean Sube
|
4de8c160c9
|
fix(build): make sure Torch is installed first in containers
|
2023-03-17 19:41:34 -05:00 |
Sean Sube
|
226710a015
|
fix(api): use exception level logs
|
2023-03-16 22:29:07 -05:00 |
Sean Sube
|
d9bae80778
|
fix(build): install wheel for basicsr
|
2023-03-16 21:46:30 -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
|
8cf6f2215d
|
fix(scripts): diff model nodes as well
|
2023-03-14 22:58:36 -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
|
cf429ad715
|
fix(tests): add reference image for UniPC scheduler
|
2023-03-11 14:43:36 -06:00 |
Sean Sube
|
a9906faec8
|
chore(release): 0.8.1
|
2023-03-11 14:32:05 -06: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 |