fix(api): run background ESRGAN on same device as GFPGAN
This commit is contained in:
parent
c7e0041229
commit
1dfd9e72ef
|
@ -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))
|
||||
|
||||
|
|
Loading…
Reference in New Issue