diff options
| author | Dean van Dugteren <31391056+deanpress@users.noreply.github.com> | 2022-12-11 16:08:51 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-11 16:08:51 +0000 |
| commit | 59c6511494c55a578eecdf71fb4590b6bd5d04a7 (patch) | |
| tree | 95efda44b427dd271c6851b288789c36e34d08f5 /modules/sd_models.py | |
| parent | 685f9631b56ff8bd43bce24ff5ce0f9a0e9af490 (diff) | |
| download | stable-diffusion-webui-gfx803-59c6511494c55a578eecdf71fb4590b6bd5d04a7.tar.gz stable-diffusion-webui-gfx803-59c6511494c55a578eecdf71fb4590b6bd5d04a7.tar.bz2 stable-diffusion-webui-gfx803-59c6511494c55a578eecdf71fb4590b6bd5d04a7.zip | |
fix: fallback model_checkpoint if it's empty
This fixes the following error when SD attempts to start with a deleted checkpoint:
```
Traceback (most recent call last):
File "D:\Web\stable-diffusion-webui\launch.py", line 295, in <module>
start()
File "D:\Web\stable-diffusion-webui\launch.py", line 290, in start
webui.webui()
File "D:\Web\stable-diffusion-webui\webui.py", line 132, in webui
initialize()
File "D:\Web\stable-diffusion-webui\webui.py", line 62, in initialize
modules.sd_models.load_model()
File "D:\Web\stable-diffusion-webui\modules\sd_models.py", line 283, in load_model
checkpoint_info = checkpoint_info or select_checkpoint()
File "D:\Web\stable-diffusion-webui\modules\sd_models.py", line 117, in select_checkpoint
checkpoint_info = checkpoints_list.get(model_checkpoint, None)
TypeError: unhashable type: 'list'
```
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 5b37f3fe..b6d75db7 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -111,6 +111,10 @@ def model_hash(filename): def select_checkpoint():
model_checkpoint = shared.opts.sd_model_checkpoint
+
+ if len(model_checkpoint) == 0:
+ model_checkpoint = shared.default_sd_model_file
+
checkpoint_info = checkpoints_list.get(model_checkpoint, None)
if checkpoint_info is not None:
return checkpoint_info
|