jhj0517 commited on
Commit
ad23aea
·
1 Parent(s): 5dba7cb

Add SRVGGNetCompact

Browse files
modules/image_restoration/real_esrgan/real_esrgan_inferencer.py CHANGED
@@ -2,14 +2,13 @@ import os.path
2
  import gradio as gr
3
  import torch
4
  from typing import Optional, Literal
5
- from realesrgan.utils import RealESRGANer
6
- from realesrgan.archs.srvgg_arch import SRVGGNetCompact
7
 
8
  from modules.utils.paths import *
9
  from modules.utils.image_helper import save_image
10
  from .model_downloader import download_resrgan_model, MODELS_REALESRGAN_URL, MODELS_REALESRGAN_SCALABILITY
11
  from wrapper.rrdb_net import RRDBNet
12
  from wrapper.real_esrganer import RealESRGANer
 
13
 
14
 
15
  class RealESRGANInferencer:
 
2
  import gradio as gr
3
  import torch
4
  from typing import Optional, Literal
 
 
5
 
6
  from modules.utils.paths import *
7
  from modules.utils.image_helper import save_image
8
  from .model_downloader import download_resrgan_model, MODELS_REALESRGAN_URL, MODELS_REALESRGAN_SCALABILITY
9
  from wrapper.rrdb_net import RRDBNet
10
  from wrapper.real_esrganer import RealESRGANer
11
+ from wrapper.srvgg_net_compact import SRVGGNetCompact
12
 
13
 
14
  class RealESRGANInferencer: