1
0
Fork 0
Commit Graph

54 Commits

Author SHA1 Message Date
Sean Sube 651acf6991
lint recent changes 2023-02-06 17:26:51 -06:00
Sean Sube 833fc5c2f8
fix(api): make sure each upscale/correction model gets its own stage 2023-02-06 17:17:43 -06:00
Sean Sube 0a9f108156
feat(api): add CodeFormer to automatic upscale 2023-02-05 10:49:20 -06:00
Sean Sube 54dd34d211
lint(api): apply black and isort style 2023-02-05 07:53:35 -06:00
Sean Sube 3637f642c6
fix(api): pass job context and device to upscaling 2023-02-04 14:52:23 -06:00
Sean Sube 6a0f881120
fix(api): pass prompt to upscaling pipeline 2023-01-31 20:41:15 -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 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 dcbd059082 feat(api): add img2img and inpaint chain stages 2023-01-28 12:42:02 -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 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 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 9d0a15900b lint(api): mark some unused vars 2023-01-22 10:08:26 -06:00
Sean Sube 9a2e7adfb8 fix(api): report accurate image size when upscaling 2023-01-19 19:42:27 -06:00
Sean Sube 4d9dc05cd0 lint(api): improve logging while upscaling 2023-01-17 22:30:25 -06:00
Sean Sube b7c85aa51b fix(api): skip upscaling if scale is 1 2023-01-16 23:01:15 -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 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
Sean Sube ee6308a091 feat(api): return all types of models 2023-01-16 20:10:52 -06:00
Sean Sube decb2813c6 feat(api): add support for Stable Diffusion models to conversion script 2023-01-16 17:48:50 -06:00
Sean Sube 5a01fe4cb0 fix(api): resolve face correction model relative to model path 2023-01-16 16:45:17 -06:00
Sean Sube 4bf68759d7 fix(api): report accurate sizes 2023-01-16 15:11:40 -06:00
Sean Sube d406cd4e99 fix(api): pass model to ONNX instances 2023-01-16 14:58:27 -06:00
Sean Sube 8d3ebede5a feat: add outscaling option 2023-01-16 14:52:56 -06:00
Sean Sube 091c4e6109 fix(api): pass upscale params when creating RESRGAN 2023-01-16 14:17:50 -06:00
Sean Sube 1f0c19af04 fix(api): get upscale params from request 2023-01-16 13:12:08 -06:00
Sean Sube 120056f878 fix(api): get ESRGAN/GFPGAN paths from server context, clean up test scripts 2023-01-16 13:02:15 -06:00
Sean Sube 0f2d6d2ec7 add subscript for fake image 2023-01-16 12:02:37 -06:00
Sean Sube 402fa4f687 return fake image throughout chain 2023-01-16 11:46:06 -06:00
Sean Sube f493246192 fix(api): correct stub method name 2023-01-16 11:45:31 -06:00
Sean Sube ba3eff5c03 fix(api): enable tiling when fixing faces after upscaling 2023-01-16 11:43:00 -06:00
Sean Sube b37f8cdd7d stub out other methods for output image 2023-01-16 11:16:06 -06:00
Sean Sube 30d474b487 wrap image output 2023-01-16 11:06:50 -06:00
Sean Sube fe657468bf fix(api): add missing params to load stub 2023-01-16 10:59:47 -06:00
Sean Sube 9519fc16e9 feat(api): add ONNX implementation of Real ESRGAN net 2023-01-16 10:55:40 -06:00
Sean Sube e59449fec1 feat(api): add conversion script for models 2023-01-16 09:57:59 -06:00