diff options
author | Aarni Koskela <akx@iki.fi> | 2023-05-29 05:54:13 +0000 |
---|---|---|
committer | Aarni Koskela <akx@iki.fi> | 2023-05-29 06:17:30 +0000 |
commit | 00dfe27f59727407c5b408a80ff2a262934df495 (patch) | |
tree | aae9d324a719c7e500f4a0c45821851bc6e41ee0 /extensions-builtin/ScuNET | |
parent | b957dcfece29c84ac0cfcd5a69475ff8684c531f (diff) | |
download | stable-diffusion-webui-gfx803-00dfe27f59727407c5b408a80ff2a262934df495.tar.gz stable-diffusion-webui-gfx803-00dfe27f59727407c5b408a80ff2a262934df495.tar.bz2 stable-diffusion-webui-gfx803-00dfe27f59727407c5b408a80ff2a262934df495.zip |
Add & use modules.errors.print_error where currently printing exception info by hand
Diffstat (limited to 'extensions-builtin/ScuNET')
-rw-r--r-- | extensions-builtin/ScuNET/scripts/scunet_model.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/extensions-builtin/ScuNET/scripts/scunet_model.py b/extensions-builtin/ScuNET/scripts/scunet_model.py index 45d9297b..dd1b822e 100644 --- a/extensions-builtin/ScuNET/scripts/scunet_model.py +++ b/extensions-builtin/ScuNET/scripts/scunet_model.py @@ -1,6 +1,5 @@ import os.path import sys -import traceback import PIL.Image import numpy as np @@ -12,6 +11,8 @@ from basicsr.utils.download_util import load_file_from_url import modules.upscaler from modules import devices, modelloader, script_callbacks from scunet_model_arch import SCUNet as net + +from modules.errors import print_error from modules.shared import opts @@ -38,8 +39,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler): scaler_data = modules.upscaler.UpscalerData(name, file, self, 4) scalers.append(scaler_data) except Exception: - print(f"Error loading ScuNET model: {file}", file=sys.stderr) - print(traceback.format_exc(), file=sys.stderr) + print_error(f"Error loading ScuNET model: {file}", exc_info=True) if add_model2: scaler_data2 = modules.upscaler.UpscalerData(self.model_name2, self.model_url2, self) scalers.append(scaler_data2) |