mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
update upscalers
This commit is contained in:
@@ -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__()
|
||||
Reference in New Issue
Block a user