update upscalers

This commit is contained in:
Vladimir Mandic
2023-09-30 09:20:22 -04:00
parent f11cf4b29a
commit d01f01c981
10 changed files with 57 additions and 40 deletions
+2 -2
View File
@@ -2,7 +2,7 @@ import PIL.Image
import numpy as np
import torch
from rich.progress import Progress, TextColumn, BarColumn, TaskProgressColumn, TimeRemainingColumn, TimeElapsedColumn
from modules import devices, script_callbacks
from modules import devices
from modules.postprocess.scunet_model_arch import SCUNet as net
from modules.shared import opts, log, console, device
from modules.upscaler import Upscaler
@@ -67,7 +67,7 @@ class UpscalerSCUNet(Upscaler):
output = E.div_(W)
return output
def do_upscale(self, img: PIL.Image.Image, selected_file): # pylint: disable=arguments-renamed
def do_upscale(self, img: PIL.Image.Image, selected_file):
devices.torch_gc()
model = self.load_model(selected_file)
if model is None:
@@ -6,7 +6,7 @@ from modules.upscaler import Upscaler, UpscalerData
class UpscalerSD(Upscaler):
def __init__(self, dirname): # pylint: disable=super-init-not-called
self.name = "StableDiffusion"
self.name = "SDUpscale"
self.user_path = dirname
if shared.backend != shared.Backend.DIFFUSERS:
super().__init__()