diff options
| author | linkoid <36754150+linkoid@users.noreply.github.com> | 2023-05-26 15:15:59 -0400 |
|---|---|---|
| committer | linkoid <36754150+linkoid@users.noreply.github.com> | 2023-05-26 15:25:31 -0400 |
| commit | 1f0fdede176989f151da6b97bd9a140b7f0af6e5 (patch) | |
| tree | e9fbaef018d301f442d8c9973261a67889939842 | |
| parent | 3829afec365b748e330da33b00a0e363f8c8ab71 (diff) | |
Show full traceback in get_sd_model()
to reveal if an error is caused by an extension
| -rw-r--r-- | modules/errors.py | 8 | ||||
| -rw-r--r-- | modules/sd_models.py | 2 |
2 files changed, 7 insertions, 3 deletions
diff --git a/modules/errors.py b/modules/errors.py index f6b80dbb..da4694f8 100644 --- a/modules/errors.py +++ b/modules/errors.py @@ -12,9 +12,13 @@ def print_error_explanation(message): print('=' * max_len, file=sys.stderr)
-def display(e: Exception, task):
+def display(e: Exception, task, *, full_traceback=False):
print(f"{task or 'error'}: {type(e).__name__}", file=sys.stderr)
- print(traceback.format_exc(), file=sys.stderr)
+ te = traceback.TracebackException.from_exception(e)
+ if full_traceback:
+ # include frames leading up to the try-catch block
+ te.stack = traceback.StackSummary(traceback.extract_stack()[:-2] + te.stack)
+ print(*te.format(), sep="", file=sys.stderr)
message = str(e)
if "copying a param with shape torch.Size([640, 1024]) from checkpoint, the shape in current model is torch.Size([640, 768])" in message:
diff --git a/modules/sd_models.py b/modules/sd_models.py index 1871cc97..3e7fc7e3 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -424,7 +424,7 @@ class SdModelData: try:
load_model()
except Exception as e:
- errors.display(e, "loading stable diffusion model")
+ errors.display(e, "loading stable diffusion model", full_traceback=True)
print("", file=sys.stderr)
print("Stable diffusion model failed to load", file=sys.stderr)
self.sd_model = None
|