Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Minor cleanup

This commit is contained in:
n0kovo 2024-01-19 00:39:14 +01:00
parent a97147bc8a
commit 2e7efe47b6
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/dat_model.py
View File

@ -1,12 +1,10 @@
import os import os
import sys
from modules import modelloader, devices from modules import modelloader, errors
from modules.shared import cmd_opts, opts from modules.shared import cmd_opts, opts
from modules.upscaler import Upscaler, UpscalerData from modules.upscaler import Upscaler, UpscalerData
from modules.upscaler_utils import upscale_with_model from modules.upscaler_utils import upscale_with_model
from icecream import ic
class UpscalerDAT(Upscaler): class UpscalerDAT(Upscaler):
def __init__(self, user_path): def __init__(self, user_path):
@ -24,13 +22,13 @@ class UpscalerDAT(Upscaler):
if model.name in opts.dat_enabled_models: if model.name in opts.dat_enabled_models:
self.scalers.append(model) self.scalers.append(model)
def do_upscale(self, img, selected_model): def do_upscale(self, img, path):
try: try:
info = self.load_model(selected_model) info = self.load_model(path)
except Exception as e: except Exception:
errors.report(f"Unable to load DAT model {path}", exc_info=True) errors.report(f"Unable to load DAT model {path}", exc_info=True)
return img return img
model_descriptor = modelloader.load_spandrel_model( model_descriptor = modelloader.load_spandrel_model(
info.local_data_path, info.local_data_path,
device=self.device, device=self.device,