Commit 0609ce06 authored by Milly's avatar Milly Committed by AUTOMATIC1111
Browse files

Removed duplicate definition model_path

parent a65a4527

modules/bsrgan_model.py

+0 −2
Original line number Diff line number Diff line
@@ -10,13 +10,11 @@ from basicsr.utils.download_util import load_file_from_url 
import modules.upscaler

from modules import devices, modelloader

from modules.bsrgan_model_arch import RRDBNet

from modules.paths import models_path





class UpscalerBSRGAN(modules.upscaler.Upscaler):

    def __init__(self, dirname):

        self.name = "BSRGAN"

        self.model_path = os.path.join(models_path, self.name)

        self.model_name = "BSRGAN 4x"

        self.model_url = "https://github.com/cszn/KAIR/releases/download/v1.0/BSRGAN.pth"

        self.user_path = dirname

modules/esrgan_model.py

+0 −2
Original line number Diff line number Diff line
@@ -7,7 +7,6 @@ from basicsr.utils.download_util import load_file_from_url 


import modules.esrgam_model_arch as arch

from modules import shared, modelloader, images, devices

from modules.paths import models_path

from modules.upscaler import Upscaler, UpscalerData

from modules.shared import opts
@@ -76,7 +75,6 @@ class UpscalerESRGAN(Upscaler): 
        self.model_name = "ESRGAN_4x"

        self.scalers = []

        self.user_path = dirname

        self.model_path = os.path.join(models_path, self.name)

        super().__init__()

        model_paths = self.find_models(ext_filter=[".pt", ".pth"])

        scalers = []

modules/ldsr_model.py

+0 −2
Original line number Diff line number Diff line
@@ -7,13 +7,11 @@ from basicsr.utils.download_util import load_file_from_url 
from modules.upscaler import Upscaler, UpscalerData

from modules.ldsr_model_arch import LDSR

from modules import shared

from modules.paths import models_path





class UpscalerLDSR(Upscaler):

    def __init__(self, user_path):

        self.name = "LDSR"

        self.model_path = os.path.join(models_path, self.name)

        self.user_path = user_path

        self.model_url = "https://heibox.uni-heidelberg.de/f/578df07c8fc04ffbadf3/?dl=1"

        self.yaml_url = "https://heibox.uni-heidelberg.de/f/31a76b13ea27482981b4/?dl=1"

modules/realesrgan_model.py

+0 −2
Original line number Diff line number Diff line
@@ -8,14 +8,12 @@ from basicsr.utils.download_util import load_file_from_url 
from realesrgan import RealESRGANer



from modules.upscaler import Upscaler, UpscalerData

from modules.paths import models_path

from modules.shared import cmd_opts, opts





class UpscalerRealESRGAN(Upscaler):

    def __init__(self, path):

        self.name = "RealESRGAN"

        self.model_path = os.path.join(models_path, self.name)

        self.user_path = path

        super().__init__()

        try:

modules/scunet_model.py

+0 −2
Original line number Diff line number Diff line
@@ -9,14 +9,12 @@ from basicsr.utils.download_util import load_file_from_url 


import modules.upscaler

from modules import devices, modelloader

from modules.paths import models_path

from modules.scunet_model_arch import SCUNet as net





class UpscalerScuNET(modules.upscaler.Upscaler):

    def __init__(self, dirname):

        self.name = "ScuNET"

        self.model_path = os.path.join(models_path, self.name)

        self.model_name = "ScuNET GAN"

        self.model_name2 = "ScuNET PSNR"

        self.model_url = "https://github.com/cszn/KAIR/releases/download/v1.0/scunet_color_real_gan.pth"

CRA Git | Maintained and supported by SUSTech CRA and CCSE