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:
|
if upsampler is None:
|
||||||
bg_upscale = upscale.rescale(upscale.outscale)
|
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))
|
face_path = path.join(ctx.model_path, "%s.pth" % (upscale.correction_model))
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue