Sean Sube
|
37dd8927bf
|
fix(api): pass device ID in provider params
|
2023-02-04 21:52:45 -06:00 |
Sean Sube
|
9d1f9412f6
|
log device options
|
2023-02-04 21:26:11 -06:00 |
Sean Sube
|
cc12cb0fcf
|
clean up lingering provider refs
|
2023-02-04 21:23:34 -06:00 |
Sean Sube
|
04a2faffd9
|
fix(api): set CUDA device in ORT session
|
2023-02-04 21:17:39 -06:00 |
Sean Sube
|
d636ce3eef
|
fix(api): avoid setting device on schedulers that do not support it
|
2023-02-04 17:21:12 -06:00 |
Sean Sube
|
13510dbacd
|
fix(api): prefer device with lower index when queue is tied
|
2023-02-04 17:03:01 -06:00 |
Sean Sube
|
d75cd50a62
|
fix non-generic types
|
2023-02-04 16:57:00 -06:00 |
Sean Sube
|
5c07fff8b4
|
lint(api): type fields in job pool class defs
|
2023-02-04 16:55:42 -06:00 |
Sean Sube
|
ae434f5c11
|
fix(api): add device index to job status endpoint
|
2023-02-04 16:55:20 -06:00 |
Sean Sube
|
ed2e15a67e
|
fix(api): exclude finished jobs when load balancing
|
2023-02-04 16:54:44 -06:00 |
Sean Sube
|
de6e3b4e27
|
only return device index
|
2023-02-04 16:40:24 -06:00 |
Sean Sube
|
bf3f227d66
|
fix(api): default to first available device when there are no other jobs
|
2023-02-04 16:37:36 -06:00 |
Sean Sube
|
efee374c16
|
feat(api): load-balance background jobs between devices
|
2023-02-04 16:22:50 -06:00 |
Sean Sube
|
f6a6c997bb
|
lint(api): log which device is assigned to which job
|
2023-02-04 15:56:32 -06:00 |
Sean Sube
|
5e0231c01b
|
feat(api): distribute jobs to devices using round-robin (#38)
|
2023-02-04 15:49:05 -06:00 |
Sean Sube
|
1e38659c80
|
extract device
|
2023-02-04 15:00:22 -06:00 |
Sean Sube
|
59f5fd41b0
|
restore provider to image params for now
|
2023-02-04 14:58:54 -06:00 |
Sean Sube
|
3637f642c6
|
fix(api): pass job context and device to upscaling
|
2023-02-04 14:52:23 -06:00 |
Sean Sube
|
8a81e8b810
|
add missing arg to chain pipeline
|
2023-02-04 14:27:27 -06:00 |
Sean Sube
|
2f1f6b3a11
|
fix chain pipeline image field names
|
2023-02-04 14:24:18 -06:00 |
Sean Sube
|
bef50d89b1
|
fix(api): move chain body to form field
|
2023-02-04 14:18:44 -06:00 |
Sean Sube
|
1de2a51db5
|
feat(api): load source and mask images for chain pipelines (#88)
|
2023-02-04 14:08:43 -06:00 |
Sean Sube
|
4e5ad54471
|
improve device logging
|
2023-02-04 14:01:30 -06:00 |
Sean Sube
|
2c9a10e2b7
|
format devices in logs
|
2023-02-04 13:56:17 -06:00 |
Sean Sube
|
a212d954eb
|
fix provider lookup
|
2023-02-04 13:52:32 -06:00 |
Sean Sube
|
98b6e4dd03
|
feat(api): add provider for each available CUDA device (#38)
|
2023-02-04 13:49:34 -06:00 |
Sean Sube
|
732aa27b92
|
fix(api): log number of background workers on startup
|
2023-02-04 12:29:30 -06:00 |
Sean Sube
|
53f492459f
|
feat(api): log errors from background jobs
|
2023-02-04 11:56:34 -06:00 |
Sean Sube
|
87bbce4fff
|
return progress through chain
|
2023-02-04 11:08:22 -06:00 |
Sean Sube
|
157ed6da70
|
feat(api): add status endpoint
|
2023-02-04 10:59:03 -06:00 |
Sean Sube
|
ac7657defd
|
add debug logs to progress
|
2023-02-04 10:50:41 -06:00 |
Sean Sube
|
294c831d02
|
feat(api): add progress to ready endpoint
|
2023-02-04 10:16:30 -06:00 |
Sean Sube
|
1491a9e1e0
|
cancel jobs correctly
|
2023-02-04 10:10:40 -06:00 |
Sean Sube
|
c3b0c5e3e3
|
make initial thread pool optional
|
2023-02-04 10:07:58 -06:00 |
Sean Sube
|
6426cff741
|
feat(api): switch to device pool for background workers
|
2023-02-04 10:06:22 -06:00 |
Sean Sube
|
ea69753bfd
|
fix(api): additional logging around model load ops
|
2023-02-02 23:34:02 -06:00 |
Sean Sube
|
aff3a3f2ee
|
lint(api): fix interpolation in some log messages
|
2023-02-02 18:52:31 -06:00 |
Sean Sube
|
a1298f1565
|
fix(api): respect model selection when running inpaint/outpaint stages
|
2023-02-02 18:51:54 -06:00 |
Sean Sube
|
406e0f0cf9
|
feat(api): provide a way to override log settings
|
2023-02-02 18:46:17 -06:00 |
Sean Sube
|
c837830043
|
feat(api): provide a way for users to add models to the convert list (#70)
|
2023-02-02 08:59:47 -06:00 |
Sean Sube
|
0050cea694
|
fix(api): use requested size for initial chain pipeline input
|
2023-02-02 08:59:08 -06:00 |
Sean Sube
|
e533dad4ba
|
fix(api): use server image format when building output name
|
2023-02-02 08:47:37 -06:00 |
Sean Sube
|
59b8055926
|
fix(api): correct import error
|
2023-02-02 08:33:33 -06:00 |
Sean Sube
|
a1ef6c4c77
|
lint(api): consolidate output fns
|
2023-02-02 08:31:35 -06:00 |
Sean Sube
|
6ce13096ed
|
feat(api): server setting to select image format
|
2023-02-02 08:19:57 -06:00 |
Sean Sube
|
83992d9193
|
fix(api): convert size to JSON
|
2023-02-01 22:37:26 -06:00 |
Sean Sube
|
2123d42c45
|
fix(api): return JSON params after image requests
|
2023-02-01 22:35:10 -06:00 |
Sean Sube
|
b1eec6907a
|
fix(api): write border and upscale to params file
|
2023-02-01 22:31:01 -06:00 |
Sean Sube
|
c19f39f9b4
|
fix(api): handle partial params when formatting output
|
2023-02-01 22:25:53 -06:00 |
Sean Sube
|
0557ab9a2e
|
lint(api): use constant for latents scale factor
|
2023-02-01 22:21:22 -06:00 |
Sean Sube
|
6697c2eb6a
|
feat(api): save image params along with image (#81)
|
2023-02-01 22:20:40 -06:00 |
Sean Sube
|
066b1a0230
|
fix(api): run garbage collection after each model change
|
2023-02-01 21:20:48 -06:00 |
Sean Sube
|
6c9b0c9c1e
|
fix(api): remove reference to undeclared var
|
2023-01-31 21:29:22 -06:00 |
Sean Sube
|
a3eed7a095
|
fix(api): drop unused parameter from inpaint pipeline
|
2023-01-31 21:14:43 -06:00 |
Sean Sube
|
6a0f881120
|
fix(api): pass prompt to upscaling pipeline
|
2023-01-31 20:41:15 -06:00 |
Sean Sube
|
33bb56dd24
|
fix(api): correctly make output directory
|
2023-01-31 19:58:07 -06:00 |
Sean Sube
|
f302be7f50
|
fix(api): correctly create model path
|
2023-01-31 19:32:34 -06:00 |
Sean Sube
|
6c22cb1cbf
|
feat(api): create model path if needed during convert script
|
2023-01-31 17:09:23 -06:00 |
Sean Sube
|
c34ddacf55
|
fix(api): pin outscale for GFPGAN to 1 to avoid sparse tiling
|
2023-01-31 17:08:30 -06:00 |
Sean Sube
|
f4fc6271bc
|
feat(api): set default platform for each container (fixes #82)
|
2023-01-31 08:45:06 -06:00 |
Sean Sube
|
1ca7edb4a8
|
fix(api): variable name in GFPGAN
|
2023-01-31 08:24:10 -06:00 |
Sean Sube
|
39f84c4a49
|
fix(api): convert images to ndarray before passing to GFPGAN
|
2023-01-31 08:16:57 -06:00 |
Sean Sube
|
d5e5588939
|
fix(api): supply missing parameter to GFPGAN
|
2023-01-30 20:54:39 -06:00 |
Sean Sube
|
ec1e3614f5
|
remove unused import
|
2023-01-29 22:07:51 -06:00 |
Sean Sube
|
301cbe8b8c
|
pass provider to upscale pipeline, clean up some debug
|
2023-01-29 18:54:35 -06:00 |
Sean Sube
|
bacce0ace9
|
testing ONNX upscaling
|
2023-01-29 18:42:05 -06:00 |
Sean Sube
|
ca613cabe1
|
more ONNX upscaling
|
2023-01-29 16:57:18 -06:00 |
Sean Sube
|
8c6d957a53
|
continue converting upscale to ONNX
|
2023-01-29 15:23:19 -06:00 |
Sean Sube
|
c905fbb728
|
feat(api): add reduce stages, noise source
|
2023-01-29 15:23:01 -06:00 |
Sean Sube
|
8d346cbed0
|
start porting SD upscale to ONNX
|
2023-01-29 13:49:30 -06:00 |
Sean Sube
|
f0c905721f
|
fix imports
|
2023-01-29 11:42:42 -06:00 |
Sean Sube
|
2033cbb601
|
fix tile latent axis
|
2023-01-29 11:39:02 -06:00 |
Sean Sube
|
50d6dbb451
|
feat(api): switch inpaint mode to upscale stage
|
2023-01-29 10:39:10 -06:00 |
Sean Sube
|
483a1bad68
|
feat(api): use per-tile latents for outpainting
|
2023-01-29 10:31:22 -06:00 |
Sean Sube
|
b1ba09af3e
|
fix(api): make sure each pipeline has its own list of stages
|
2023-01-29 00:25:49 -06:00 |
Sean Sube
|
7083505483
|
fix(api): preserve new pixels after outpainting
|
2023-01-29 00:16:09 -06:00 |
Sean Sube
|
20beff839e
|
fix spiral, override prompt
|
2023-01-29 00:07:12 -06:00 |
Sean Sube
|
6f1ffaa043
|
fix(api): convert spiral grid coordinates to offsets
|
2023-01-28 23:58:04 -06:00 |
Sean Sube
|
03d88dff5f
|
log pipeline stage count, add outpaint example
|
2023-01-28 23:50:41 -06:00 |
Sean Sube
|
a4d3f18a48
|
feat(api): implement spiral grid for outpainting
|
2023-01-28 23:46:36 -06:00 |
Sean Sube
|
680adc70ea
|
feat(api): save output of each stage in debug mode
|
2023-01-28 23:08:56 -06:00 |
Sean Sube
|
8f1cbc83f8
|
feat(api): parse named tile sizes
|
2023-01-28 23:06:25 -06:00 |
Sean Sube
|
db9189fd3d
|
get upscaling model from stage params
|
2023-01-28 22:52:39 -06:00 |
Sean Sube
|
18803db6ed
|
fix(api): parse border and upscale in chain stages
|
2023-01-28 22:48:53 -06:00 |
Sean Sube
|
151ebff237
|
feat(api): parse chain pipeline from request
|
2023-01-28 22:31:34 -06:00 |
Sean Sube
|
c7a6ec45d8
|
fix(api): avoid circular deps in diffusion pipeline cache
|
2023-01-28 20:15:39 -06:00 |
Sean Sube
|
3359138a83
|
clean up logging, use extra args
|
2023-01-28 17:15:42 -06:00 |
Sean Sube
|
4547bcef50
|
feat(api): switch to python logging
|
2023-01-28 17:09:19 -06:00 |
Sean Sube
|
bb3a7dc0e9
|
log pipeline timing, add common sizes
|
2023-01-28 14:56:06 -06:00 |
Sean Sube
|
8d57d113cd
|
feat(api): add s3 upload chain stage
|
2023-01-28 12:42:22 -06:00 |
Sean Sube
|
dcbd059082
|
feat(api): add img2img and inpaint chain stages
|
2023-01-28 12:42:02 -06:00 |
Sean Sube
|
4188b019a1
|
test multiple chained upscaling stages
|
2023-01-28 10:26:04 -06:00 |
Sean Sube
|
ba3e450d43
|
fix scale for outpaint
|
2023-01-28 09:55:47 -06:00 |
Sean Sube
|
2c8fec0e53
|
size outpaint mask correctly
|
2023-01-28 09:38:25 -06:00 |
Sean Sube
|
d6193a9a47
|
fix(api): tile mask during outpaint stage
|
2023-01-28 09:23:18 -06:00 |
Sean Sube
|
d89437b11e
|
persist to output dir, do not tile output
|
2023-01-28 09:15:02 -06:00 |
Sean Sube
|
779457b546
|
feat(api): add save-to-disk stage
|
2023-01-28 09:08:59 -06:00 |
Sean Sube
|
ce6cf08684
|
fix outpaint and txt2img stages
|
2023-01-28 08:44:24 -06:00 |
Sean Sube
|
658e12266f
|
pass size and initial image to test pipeline
|
2023-01-28 08:40:29 -06:00 |
Sean Sube
|
9f76226ba6
|
use correct generator
|
2023-01-28 08:38:48 -06:00 |
Sean Sube
|
5119a982db
|
feat(api): synthesize a mask for outpaint stages
|
2023-01-28 08:37:17 -06:00 |
Sean Sube
|
4579e96cc1
|
feat(api): add inpaint as a chain stage
|
2023-01-28 08:19:40 -06:00 |
Sean Sube
|
c7fa50a028
|
fix cache globals
|
2023-01-28 00:08:52 -06:00 |
Sean Sube
|
ee495a8d93
|
fix(api): correct caching for upscaling models
|
2023-01-28 00:05:37 -06:00 |
Sean Sube
|
56a4acee2a
|
fix(api): split up and cache upscaling and correction stages
|
2023-01-27 23:28:14 -06:00 |
Sean Sube
|
caafc9ebc9
|
lint(api): make modules for chain pipeline and params
|
2023-01-27 22:48:06 -06:00 |
Sean Sube
|
bcaf0f73e6
|
feat(api): implement upscaling and correction as a chain pipeline
|
2023-01-27 17:38:21 -06:00 |
Sean Sube
|
9b37c174d1
|
fix parameter order for chain pipeline
|
2023-01-27 17:26:48 -06:00 |
Sean Sube
|
bfd8e0a739
|
fix import
|
2023-01-27 17:11:27 -06:00 |
Sean Sube
|
71ff3bb1c4
|
feat(api): start implementing chain pipelines
|
2023-01-27 17:08:36 -06:00 |
Sean Sube
|
260f7a29f4
|
fix model name
|
2023-01-26 23:15:43 -06:00 |
Sean Sube
|
b971b659e3
|
use a torch rng
|
2023-01-26 23:15:24 -06:00 |
Sean Sube
|
670bc1ced5
|
improve logging, provide rng within filter
|
2023-01-26 22:58:36 -06:00 |
Sean Sube
|
936ebba510
|
attempt to run SD upscaling in smaller tiles
|
2023-01-26 22:44:20 -06:00 |
Sean Sube
|
fb5c46d90c
|
load additional components for SD upscaling
|
2023-01-25 21:29:18 -06:00 |
Sean Sube
|
819af824b0
|
feat(api): initial support for Stable Diffusion upscaling (#66)
|
2023-01-25 21:04:09 -06:00 |
Sean Sube
|
483b8e3f19
|
lint(api): consolidate upscale/correction logic
|
2023-01-25 20:31:39 -06:00 |
Sean Sube
|
67d51a96e3
|
feat(api): add a way for the server to disable certain platforms (#83)
|
2023-01-24 23:19:57 -06:00 |
Sean Sube
|
6b2ae2aeab
|
lint(api): suppress another common warning during conversion (fixes #75)
|
2023-01-23 21:40:11 -06:00 |
Sean Sube
|
b62c7d3742
|
fix(api): return structured error when image parameters are missing (fixes #76)
|
2023-01-22 22:38:03 -06:00 |
Sean Sube
|
b2e7ad599f
|
fix(api): check if output file exists for ready endpoint (#57)
|
2023-01-22 22:25:00 -06:00 |
Sean Sube
|
a5d3ffcc73
|
fix(api): use correct coordinate system for outpainting
|
2023-01-22 22:09:39 -06:00 |
Sean Sube
|
9bff64c7c9
|
fix(api): correct name for kwargs
|
2023-01-22 17:25:29 -06:00 |
Sean Sube
|
f319e6a49b
|
fix(api): pass hardware platform to upscaling pipeline (#77)
|
2023-01-22 16:35:53 -06:00 |
Sean Sube
|
1b2618c0f7
|
lint(api): suppress some known warnings during model conversion (#75)
|
2023-01-22 15:45:09 -06:00 |
Sean Sube
|
2921eba1f6
|
fix(api): handle parameters correctly when list of valid values is empty (fixes #72)
|
2023-01-22 15:44:35 -06:00 |
Sean Sube
|
876b54a7a8
|
fix(api): validate request params better, esp model path
|
2023-01-22 13:48:14 -06:00 |
Sean Sube
|
9d0a15900b
|
lint(api): mark some unused vars
|
2023-01-22 10:08:26 -06:00 |
Sean Sube
|
dddadfc9a2
|
fix(api): explicitly delete pipeline results after saving
|
2023-01-21 22:28:13 -06:00 |
Sean Sube
|
9e26ee5b85
|
fix(api): unload old model before loading next one
|
2023-01-21 22:09:32 -06:00 |
Sean Sube
|
abc1ae5112
|
fix(api): put conversion RNG on training device (#67)
|
2023-01-21 19:59:58 -06:00 |
Sean Sube
|
cb005d3b5d
|
fix: move output path out of API route (for #7)
|
2023-01-21 19:44:54 -06:00 |
Sean Sube
|
facd69f452
|
fix(api): actually return the filtered list of platforms
|
2023-01-21 19:41:43 -06:00 |
Sean Sube
|
c768cd8f42
|
fix(api): filter platforms based on available providers (fixes #69)
|
2023-01-21 19:40:10 -06:00 |
Sean Sube
|
8c9c99eeb5
|
fix(api): use training device when loading Real ESRGAN model (#67)
|
2023-01-21 17:55:55 -06:00 |
Sean Sube
|
505cacfbc2
|
fix(api): leave pipelines on default device unless specified
|
2023-01-21 17:04:48 -06:00 |
Sean Sube
|
a868c8cf6b
|
feat(api): backend support for multiple GPUs in diffusion pipelines
|
2023-01-21 15:17:33 -06:00 |
Sean Sube
|
45a3ddc2a9
|
feat(api): add option for HuggingFace token in convert script
|
2023-01-21 15:08:19 -06:00 |
Sean Sube
|
e0834110fc
|
fix(api): add extra models to convert script
|
2023-01-21 08:52:35 -06:00 |
Sean Sube
|
be16f33151
|
fix(api): pass txt2img dimensions in correct order
|
2023-01-19 22:07:31 -06:00 |
Sean Sube
|
86fb2ae28e
|
fix(api): generate correct latents for non-square images
|
2023-01-19 22:00:58 -06:00 |
Sean Sube
|
9a10261273
|
put functions in order
|
2023-01-19 21:03:29 -06:00 |
Sean Sube
|
4a3bb97342
|
fix(api): run GC after changing pipeline (#58)
|
2023-01-19 19:46:36 -06:00 |
Sean Sube
|
9a2e7adfb8
|
fix(api): report accurate image size when upscaling
|
2023-01-19 19:42:27 -06:00 |
Sean Sube
|
3bcd7a8156
|
feat: add ROCm provider to list (#10)
|
2023-01-19 17:38:26 -06:00 |
Sean Sube
|
4fbf046e28
|
lint(api): remove unnecessary pass
|
2023-01-19 08:34:38 -06:00 |
Sean Sube
|
3679735d86
|
feat: add fill color control to inpaint
|
2023-01-18 08:41:02 -06:00 |
Sean Sube
|
08dbc0c738
|
fix(api): check image size before blending
|
2023-01-18 08:29:55 -06:00 |
Sean Sube
|
4d9dc05cd0
|
lint(api): improve logging while upscaling
|
2023-01-17 22:30:25 -06:00 |
Sean Sube
|
cf6a151548
|
fix(api): pass image size to upscale job
|
2023-01-17 21:53:44 -06:00 |