Commit 2e7efe47 authored by n0kovo's avatar n0kovo

Minor cleanup

parent a97147bc
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,10 +22,10 @@ class UpscalerDAT(Upscaler): ...@@ -24,10 +22,10 @@ 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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment