Commit 1f0fdede authored by linkoid's avatar linkoid
Browse files

Show full traceback in get_sd_model()

to reveal if an error is caused by an extension
parent 3829afec
Loading
Loading
Loading
Loading
+6 −2
Original line number Diff line number Diff line
@@ -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:
+1 −1
Original line number Diff line number Diff line
@@ -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