apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): load Real ESRGAN PTH from cache

This commit is contained in:
Sean Sube 2023-02-13 18:21:05 -06:00
parent 8dc1ea7f1f
commit 51d53597c0
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/upscale_resrgan.py
View File

@ -71,7 +71,7 @@ def load_resrgan(
logger.debug("loading Real ESRGAN upscale model from %s", model_path) logger.debug("loading Real ESRGAN upscale model from %s", model_path)
# TODO: shouldn't need the PTH file # TODO: shouldn't need the PTH file
model_path_pth = path.join(server.model_path, "%s.pth" % params.upscale_model) model_path_pth = path.join(server.model_path, ".cache", ("%s.pth" % params.upscale_model))
upsampler = RealESRGANer( upsampler = RealESRGANer(
scale=params.scale, scale=params.scale,
model_path=model_path_pth, model_path=model_path_pth,