diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-11 07:24:56 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-11 07:24:56 +0000 |
| commit | 4fdacd31e48c6a7a35c1c25c559932585e8addde (patch) | |
| tree | 71c417c1ed2fad310a4dc41450bb64357c3631be /modules/sd_models.py | |
| parent | 954091697fce7a1b7997d5f3d73551f793f6bebc (diff) | |
| download | stable-diffusion-webui-gfx803-4fdacd31e48c6a7a35c1c25c559932585e8addde.tar.gz stable-diffusion-webui-gfx803-4fdacd31e48c6a7a35c1c25c559932585e8addde.tar.bz2 stable-diffusion-webui-gfx803-4fdacd31e48c6a7a35c1c25c559932585e8addde.zip | |
possible fix for fallback for fast model creation from config
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index b5bc12f0..a0a8a909 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -337,6 +337,9 @@ def load_model(checkpoint_info=None): with sd_disable_initialization.DisableInitialization():
sd_model = instantiate_from_config(sd_config.model)
except Exception as e:
+ pass
+
+ if sd_model is None:
print('Failed to create model quickly; will retry using slow method.', file=sys.stderr)
sd_model = instantiate_from_config(sd_config.model)
|