1
0
Fork 0
Commit Graph

357 Commits

Author SHA1 Message Date
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 88e4f74efe fix(gui): load remaining defaults from server params 2023-01-27 17:27:11 -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 99897af004 lint(api): put params in order 2023-01-24 22:18:44 -06:00
Sean Sube 1a28f9bf34
chore(release): 0.5.0 2023-01-24 08:44:52 -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 5c026c43d4 fix(build): install prebuilt ROCm ORT 2023-01-23 17:17:10 -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 a8bc371478 feat(build): compile ONNX runtime with ROCm support 2023-01-22 20:49:08 -06:00
Sean Sube 348a4e2405 fix(build): correct venv path in containers 2023-01-22 19:08:41 -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 88f9b624ec fix(build): add launcher script and use for API images 2023-01-21 22:34:44 -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 5286c4f596 fix(build): add params file to other images 2023-01-21 17:55:30 -06:00
Sean Sube 505cacfbc2 fix(api): leave pipelines on default device unless specified 2023-01-21 17:04:48 -06:00
Sean Sube 52484e6e1f fix(build): add cv2 deps to CUDA and ROCm images 2023-01-21 17:01:02 -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 a8769a5919 feat(build): run convert script when container starts 2023-01-21 15:10:59 -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 77d68bf6ac fix(build): add cv2 deps to CPU image 2023-01-21 14:43:36 -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 b18567ca43 feat(build): add DirectML and ROCm images 2023-01-19 17:16:59 -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 be3a17b2ff feat: add version check to parameters 2023-01-17 23:06:04 -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
Sean Sube 634d2e8ce6 fix(api): use correct base path for debug images 2023-01-17 17:50:56 -06:00
Sean Sube 5ba752e526 feat(api): add strength param to inpaint, remove same from upscale 2023-01-17 17:50:36 -06:00
Sean Sube b496e7121c fix(api): premultiply noise before compositing 2023-01-17 17:49:46 -06:00
Sean Sube 4aeee60b19 feat: add upscaling tab and endpoint 2023-01-16 23:45:54 -06:00
Sean Sube b7c85aa51b fix(api): skip upscaling if scale is 1 2023-01-16 23:01:15 -06:00
Sean Sube 227056d976 fix(api): correct denoise max, add missing face strength param 2023-01-16 22:55:54 -06:00
Sean Sube 0e27cc830d fix: send missing model params, add face strength control 2023-01-16 22:54:01 -06:00
Sean Sube ac3ef0ebbf mostly fix face correction 2023-01-16 22:35:34 -06:00
Sean Sube c6ce43b00d back to single GFPGAN, fix some paths 2023-01-16 21:55:54 -06:00
Sean Sube 03e06193eb attempt to fix face correction at various scales 2023-01-16 21:36:50 -06:00
Sean Sube 073ff8e02f fix(api): use correct scale for background correction 2023-01-16 21:12:17 -06:00