From c943a55a44c2630d8ec7fb6924d54ec35e38e8cb Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 17 Mar 2023 23:06:39 -0500 Subject: [PATCH] fix(api): move imports of patches modules into function --- api/onnx_web/convert/upscale_resrgan.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/convert/upscale_resrgan.py b/api/onnx_web/convert/upscale_resrgan.py index 1e3cb721..ef44c790 100644 --- a/api/onnx_web/convert/upscale_resrgan.py +++ b/api/onnx_web/convert/upscale_resrgan.py @@ -2,8 +2,6 @@ from logging import getLogger from os import path import torch -from basicsr.archs.rrdbnet_arch import RRDBNet -from realesrgan.archs.srvgg_arch import SRVGGNetCompact from torch.onnx import export from .utils import ConversionContext, ModelDict @@ -19,6 +17,9 @@ def convert_upscale_resrgan( model: ModelDict, source: str, ): + from basicsr.archs.rrdbnet_arch import RRDBNet + from realesrgan.archs.srvgg_arch import SRVGGNetCompact + name = model.get("name") source = source or model.get("source") scale = model.get("scale")