fix(api): move imports of patches modules into function

ssube · Mar 18, 2023 · c943a55 · c943a55
1 parent fc9245f
commit c943a55
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/api/onnx_web/convert/upscale_resrgan.py b/api/onnx_web/convert/upscale_resrgan.py
@@ -2,8 +2,6 @@
 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")