aboutsummaryrefslogtreecommitdiffstats
path: root/extensions-builtin/LDSR/scripts/ldsr_model.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-06-27 06:02:38 +0000
committerGitHub <noreply@github.com>2023-06-27 06:02:38 +0000
commit58a9a261c4feb729717c386fea70f8ea985c6722 (patch)
treeebc822c17a889034672297d7450477f7c945cbb7 /extensions-builtin/LDSR/scripts/ldsr_model.py
parent373ff5a217eca33607abb692b9ebfa38abb7fe33 (diff)
parent2c43dd766da52d2ccaaa78d8f265a6a4aa33b9df (diff)
downloadstable-diffusion-webui-gfx803-58a9a261c4feb729717c386fea70f8ea985c6722.tar.gz
stable-diffusion-webui-gfx803-58a9a261c4feb729717c386fea70f8ea985c6722.tar.bz2
stable-diffusion-webui-gfx803-58a9a261c4feb729717c386fea70f8ea985c6722.zip
Merge branch 'dev' into meta_class
Diffstat (limited to 'extensions-builtin/LDSR/scripts/ldsr_model.py')
-rw-r--r--extensions-builtin/LDSR/scripts/ldsr_model.py8
1 files changed, 2 insertions, 6 deletions
diff --git a/extensions-builtin/LDSR/scripts/ldsr_model.py b/extensions-builtin/LDSR/scripts/ldsr_model.py
index c4da79f3..dbd6d331 100644
--- a/extensions-builtin/LDSR/scripts/ldsr_model.py
+++ b/extensions-builtin/LDSR/scripts/ldsr_model.py
@@ -1,12 +1,10 @@
import os
-import sys
-import traceback
from basicsr.utils.download_util import load_file_from_url
from modules.upscaler import Upscaler, UpscalerData
from ldsr_model_arch import LDSR
-from modules import shared, script_callbacks
+from modules import shared, script_callbacks, errors
import sd_hijack_autoencoder # noqa: F401
import sd_hijack_ddpm_v1 # noqa: F401
@@ -51,10 +49,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)
+ errors.report("Error importing LDSR", exc_info=True)
return None
def do_upscale(self, img, path):