import sys import textwrap import traceback def print_error( message: str, *, exc_info: bool = False, ) -> None: """ Print an error message to stderr, with optional traceback. """ for line in message.splitlines(): print("***", line, file=sys.stderr) if exc_info: print(textwrap.indent(traceback.format_exc(), " "), file=sys.stderr) print("---") def print_error_explanation(message): 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): print(f"{task or 'error'}: {type(e).__name__}", 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: 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): 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)