From 7ae3dc286623771c2925d8d91dbca736920757e8 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sun, 18 Sep 2022 23:52:01 +0300
Subject: [PATCH] display a more informative message when a checkpoint is not
 found

---
 modules/sd_models.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 4bd70fc5..dc81b0dc 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -76,8 +76,11 @@ def select_checkpoint():
         return checkpoint_info
 
     if len(checkpoints_list) == 0:
-        print(f"Checkpoint {model_checkpoint} not found and no other checkpoints found", file=sys.stderr)
-        return None
+        print(f"No checkpoints found. When searching for checkpoints, looked at:", file=sys.stderr)
+        print(f" - file {os.path.abspath(shared.cmd_opts.ckpt)}", file=sys.stderr)
+        print(f" - directory {os.path.abspath(shared.cmd_opts.ckpt_dir)}", file=sys.stderr)
+        print(f"Can't run without a checkpoint. Find and place a .ckpt file into any of those locations. The program will exit.", file=sys.stderr)
+        exit(1)
 
     checkpoint_info = next(iter(checkpoints_list.values()))
     if model_checkpoint is not None: