From 1dfd9e72ef8a7fb847d8a36164ad7020616c1354 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 6 Feb 2023 08:10:36 -0600 Subject: [PATCH] fix(api): run background ESRGAN on same device as GFPGAN --- api/onnx_web/chain/correct_gfpgan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/chain/correct_gfpgan.py b/api/onnx_web/chain/correct_gfpgan.py index 203c8375..e25ac21e 100644 --- a/api/onnx_web/chain/correct_gfpgan.py +++ b/api/onnx_web/chain/correct_gfpgan.py @@ -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))