diff options
author | linkoid <36754150+linkoid@users.noreply.github.com> | 2023-05-26 19:08:53 +0000 |
---|---|---|
committer | linkoid <36754150+linkoid@users.noreply.github.com> | 2023-05-26 19:08:53 +0000 |
commit | 3829afec365b748e330da33b00a0e363f8c8ab71 (patch) | |
tree | dfa087d6a1827cb0ceee82cf7335e44e2622c1d0 /modules/sd_models.py | |
parent | a6e653be26cc05f4438145fa0082816e9fbbf5fc (diff) | |
download | stable-diffusion-webui-gfx803-3829afec365b748e330da33b00a0e363f8c8ab71.tar.gz stable-diffusion-webui-gfx803-3829afec365b748e330da33b00a0e363f8c8ab71.tar.bz2 stable-diffusion-webui-gfx803-3829afec365b748e330da33b00a0e363f8c8ab71.zip |
Remove exit() from select_checkpoint()
Raising a FileNotFoundError instead.
Diffstat (limited to 'modules/sd_models.py')
-rw-r--r-- | modules/sd_models.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 91b3eb11..1871cc97 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:
|