fix(api): load Real ESRGAN PTH from cache

ssube · Feb 14, 2023 · 51d5359 · 51d5359
1 parent 8dc1ea7
commit 51d5359
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,