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

View File

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