1
0
Fork 0
Commit Graph

240 Commits

Author SHA1 Message Date
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
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 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
Sean Sube b09feda474 fix(api): sort models without discarding 2023-01-16 20:58:08 -06:00
Sean Sube 353a65513f fix(api): copy checkpoints into correct location, handle more models 2023-01-16 20:53:12 -06:00
Sean Sube 1db5ebec84 syntax fix 2023-01-16 20:40:01 -06:00
Sean Sube d52c22e58b feat: add additional Real ESRGAN models, strings for them 2023-01-16 20:38:16 -06:00
Sean Sube 33eebacc0c lint(api): return models in order 2023-01-16 20:37:59 -06:00
Sean Sube c0ca7cf62f fix(api): dedupe models after removing extension 2023-01-16 20:28:29 -06:00
Sean Sube 4472a6fd24 fix(api): trim model names relative to model path 2023-01-16 20:25:09 -06:00
Sean Sube 23a9d5afb7 list models correctly 2023-01-16 20:17:01 -06:00
Sean Sube 0273dea2a6 fix(api): join globs to avoid py 3.10-only args 2023-01-16 20:14:11 -06:00
Sean Sube ee6308a091 feat(api): return all types of models 2023-01-16 20:10:52 -06:00
Sean Sube dba6113c09 fix(api): include model scale 2023-01-16 20:10:29 -06:00
Sean Sube 556d5b84d6 fix(api): add an option to skip certain models during conversion 2023-01-16 17:55:58 -06:00
Sean Sube dab2cc7752 lint(api): split up face correction and upscaling models 2023-01-16 17:51:32 -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 4d0898a52c feat(api): start adding model sources to convert script 2023-01-16 17:08:59 -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 1d90f9b53c fix class method 2023-01-16 16:40:59 -06:00
Sean Sube 4809e00982 fix(api): get all server paths from environ 2023-01-16 16:40:32 -06:00
Sean Sube c98c0ff4dd lint(api): remove global paths entirely 2023-01-16 16:39:30 -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 adeada19e5 fix exports 2023-01-16 14:54:45 -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 5fded3cf40 remove unused param from output 2023-01-16 10:03:30 -06:00
Sean Sube e59449fec1 feat(api): add conversion script for models 2023-01-16 09:57:59 -06:00
Sean Sube 1283bc3d3f fix(api): look up noise coordinates correctly 2023-01-16 07:49:25 -06:00
Sean Sube 604cdf39b0 lint(api): helper to get borders from request 2023-01-16 07:45:50 -06:00
Sean Sube 53fcc0b91f lint(api): remove output path, join from context 2023-01-16 07:42:10 -06:00
Sean Sube c8944a2be8 fix path 2023-01-16 07:36:35 -06:00
Sean Sube 7c9dbf2c7e remove unused fields from context 2023-01-16 07:34:37 -06:00
Sean Sube 8a97cbebb6 remove invalid model path call 2023-01-16 07:33:53 -06:00
Sean Sube 7c52743aaa remove invalid imports 2023-01-16 07:33:22 -06:00
Sean Sube 13a4fa2278 lint(api): encapsulate paths in server context class 2023-01-16 07:31:42 -06:00
Sean Sube a76793d105 fix(api): correct output paths, read strength from params 2023-01-15 20:00:26 -06:00
Sean Sube f9c33486c1 fix latents for txt2img 2023-01-15 19:49:40 -06:00
Sean Sube 7d8ea23166 fix param hashing 2023-01-15 19:48:43 -06:00
Sean Sube bdc8e277fe lint(api): clean up output file hash stuff 2023-01-15 19:47:57 -06:00
Sean Sube e43238b327 fix init imports 2023-01-15 19:34:51 -06:00
Sean Sube 44c81d966f lint(api): finish splitting up modules 2023-01-15 19:33:40 -06:00
Sean Sube ff70d36a05 lint(api): add class for params, size, other common data 2023-01-15 19:14:58 -06:00
Sean Sube 1146118b1a lint(api): split up modules 2023-01-15 18:54:20 -06:00
Sean Sube c7c3645466 lint(api): split out pipeline code 2023-01-15 18:46:00 -06:00
Sean Sube 806503c709 fix(api): load upscaling model from models dir 2023-01-15 18:41:58 -06:00
Sean Sube 45d65d1342 fix(api): convert back to PIL after upscaling 2023-01-15 18:25:12 -06:00
Sean Sube 1fe6fa91fb fix(api): convert image to numpy before upscaling 2023-01-15 18:13:28 -06:00