Merge pull request #10739 from linkoid/fix-ui-debug-mode-exit

    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:

def select_checkpoint():

    """Raises `FileNotFoundError` if no checkpoints are found."""

    model_checkpoint = shared.opts.sd_model_checkpoint

    checkpoint_info = checkpoint_alisases.get(model_checkpoint, None)

        return checkpoint_info

    if len(checkpoints_list) == 0:

        print("No checkpoints found. When searching for checkpoints, looked at:", file=sys.stderr)

        error_message = "No checkpoints found. When searching for checkpoints, looked at:"

        if shared.cmd_opts.ckpt is not None:

            print(f" - file {os.path.abspath(shared.cmd_opts.ckpt)}", file=sys.stderr)

        print(f" - directory {model_path}", file=sys.stderr)

            error_message += f"\n - file {os.path.abspath(shared.cmd_opts.ckpt)}"

        error_message += f"\n - directory {model_path}"

        if shared.cmd_opts.ckpt_dir is not None:

            print(f" - directory {os.path.abspath(shared.cmd_opts.ckpt_dir)}", file=sys.stderr)

        print("Can't run without a checkpoint. Find and place a .ckpt or .safetensors file into any of those locations. The program will exit.", file=sys.stderr)

        exit(1)

            error_message += f"\n - directory {os.path.abspath(shared.cmd_opts.ckpt_dir)}"

        error_message += "Can't run without a checkpoint. Find and place a .ckpt or .safetensors file into any of those locations."

        raise FileNotFoundError(error_message)

    checkpoint_info = next(iter(checkpoints_list.values()))

    if model_checkpoint is not None:

                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