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

Fix --ui-debug-mode exit

1 files changed, 8 insertions, 7 deletions

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 835bc016..232eb9c4 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -164,6 +164,7 @@ def model_hash(filename):
 

 

 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)

@@ -171,14 +172,14 @@ def select_checkpoint():
         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:

@@ -423,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