1
0
Fork 0

switch to internal RRDBNet with x2 support

This commit is contained in:
Sean Sube 2023-12-29 08:50:27 -06:00
parent 1035915d36
commit da6beb31b2
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 2 deletions

View File

@ -5,6 +5,7 @@ from re import compile
import torch
from torch.onnx import export
from ...models.rrdb import RRDBNet
from ...models.srvgg import SRVGGNetCompact
from ..utils import ConversionContext, ModelDict
@ -61,8 +62,6 @@ def convert_upscale_resrgan(
model: ModelDict,
source: str,
):
from basicsr.archs.rrdbnet_arch import RRDBNet
name = model.get("name")
source = source or model.get("source")
scale = model.get("scale")