From 51d53597c0acd164417cc14d6699c3b32a0788fe Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 13 Feb 2023 18:21:05 -0600 Subject: [PATCH] fix(api): load Real ESRGAN PTH from cache --- api/onnx_web/chain/upscale_resrgan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/chain/upscale_resrgan.py b/api/onnx_web/chain/upscale_resrgan.py index 1dee1551..0d8f7b2c 100644 --- a/api/onnx_web/chain/upscale_resrgan.py +++ b/api/onnx_web/chain/upscale_resrgan.py @@ -71,7 +71,7 @@ def load_resrgan( logger.debug("loading Real ESRGAN upscale model from %s", model_path) # 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( scale=params.scale, model_path=model_path_pth,