apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): run background ESRGAN on same device as GFPGAN

This commit is contained in:
Sean Sube 2023-02-06 08:10:36 -06:00
parent c7e0041229
commit 1dfd9e72ef
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/correct_gfpgan.py
View File

@ -27,7 +27,7 @@ def load_gfpgan(
if upsampler is None:
bg_upscale = upscale.rescale(upscale.outscale)
upsampler = load_resrgan(ctx, bg_upscale)
upsampler = load_resrgan(ctx, bg_upscale, device)
face_path = path.join(ctx.model_path, "%s.pth" % (upscale.correction_model))