|
import sys |
|
import textwrap |
|
import traceback |
|
|
|
|
|
exception_records = [] |
|
|
|
|
|
def record_exception(): |
|
_, e, tb = sys.exc_info() |
|
if e is None: |
|
return |
|
|
|
if exception_records and exception_records[-1] == e: |
|
return |
|
|
|
exception_records.append((e, tb)) |
|
|
|
if len(exception_records) > 5: |
|
exception_records.pop(0) |
|
|
|
|
|
def report(message: str, *, exc_info: bool = False) -> None: |
|
""" |
|
Print an error message to stderr, with optional traceback. |
|
""" |
|
|
|
record_exception() |
|
|
|
for line in message.splitlines(): |
|
print("***", line, file=sys.stderr) |
|
if exc_info: |
|
print(textwrap.indent(traceback.format_exc(), " "), file=sys.stderr) |
|
print("---", file=sys.stderr) |
|
|
|
|
|
def print_error_explanation(message): |
|
record_exception() |
|
|
|
lines = message.strip().split("\n") |
|
max_len = max([len(x) for x in lines]) |
|
|
|
print('=' * max_len, file=sys.stderr) |
|
for line in lines: |
|
print(line, file=sys.stderr) |
|
print('=' * max_len, file=sys.stderr) |
|
|
|
|
|
def display(e: Exception, task, *, full_traceback=False): |
|
record_exception() |
|
|
|
print(f"{task or 'error'}: {type(e).__name__}", file=sys.stderr) |
|
te = traceback.TracebackException.from_exception(e) |
|
if full_traceback: |
|
|
|
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: |
|
print_error_explanation(""" |
|
The most likely cause of this is you are trying to load Stable Diffusion 2.0 model without specifying its config file. |
|
See https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Features#stable-diffusion-20 for how to solve this. |
|
""") |
|
|
|
|
|
already_displayed = {} |
|
|
|
|
|
def display_once(e: Exception, task): |
|
record_exception() |
|
|
|
if task in already_displayed: |
|
return |
|
|
|
display(e, task) |
|
|
|
already_displayed[task] = 1 |
|
|
|
|
|
def run(code, task): |
|
try: |
|
code() |
|
except Exception as e: |
|
display(task, e) |
|
|
