Merge pull request #7470 from cbrownstein-lambda/update-error-message-no-checkpoint

Update error message WRT missing checkpoint file

1 files changed, 1 insertions, 1 deletions

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 6c6bb571..0e61d323 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -161,7 +161,7 @@ def select_checkpoint():
         print(f" - directory {model_path}", file=sys.stderr)

         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 file into any of those locations. The program will exit.", 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)

 

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