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/LDSR | |
| 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/LDSR')
| -rw-r--r-- | extensions-builtin/LDSR/scripts/ldsr_model.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/extensions-builtin/LDSR/scripts/ldsr_model.py b/extensions-builtin/LDSR/scripts/ldsr_model.py index c4da79f3..95f1669d 100644 --- a/extensions-builtin/LDSR/scripts/ldsr_model.py +++ b/extensions-builtin/LDSR/scripts/ldsr_model.py @@ -1,9 +1,8 @@ import os -import sys -import traceback from basicsr.utils.download_util import load_file_from_url +from modules.errors import print_error from modules.upscaler import Upscaler, UpscalerData from ldsr_model_arch import LDSR from modules import shared, script_callbacks @@ -51,10 +50,8 @@ class UpscalerLDSR(Upscaler): try: return LDSR(model, yaml) - except Exception: - print("Error importing LDSR:", file=sys.stderr) - print(traceback.format_exc(), file=sys.stderr) + print_error("Error importing LDSR", exc_info=True) return None def do_upscale(self, img, path): |