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

ssube · Feb 6, 2023 · 1dfd9e7 · 1dfd9e7
1 parent c7e0041
commit 1dfd9e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))