aboutsummaryrefslogtreecommitdiffstats
path: root/extensions-builtin/LDSR/scripts/ldsr_model.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-05-31 16:16:14 +0000
committerGitHub <noreply@github.com>2023-05-31 16:16:14 +0000
commitd9bd7ada7675d3903181a6a65ba625d82cf88443 (patch)
tree30470f64d84cadc4140023c0bf34c35bb68f79b0 /extensions-builtin/LDSR/scripts/ldsr_model.py
parent78a602ae8c006077ca93913576335a3a33dba7cb (diff)
parent52b8752e6201e24c783f674f8dc0681027e10ea9 (diff)
downloadstable-diffusion-webui-gfx803-d9bd7ada7675d3903181a6a65ba625d82cf88443.tar.gz
stable-diffusion-webui-gfx803-d9bd7ada7675d3903181a6a65ba625d82cf88443.tar.bz2
stable-diffusion-webui-gfx803-d9bd7ada7675d3903181a6a65ba625d82cf88443.zip
Merge pull request #10820 from akx/report-error
Add & use modules.errors.print_error
Diffstat (limited to 'extensions-builtin/LDSR/scripts/ldsr_model.py')
-rw-r--r--extensions-builtin/LDSR/scripts/ldsr_model.py7
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):