diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-10 14:46:59 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-10 14:46:59 +0000 |
| commit | 0f8603a55988d22616b17140e6c4a7e9d0736af5 (patch) | |
| tree | 495f1902524aaa4112883808a4a27d0f5a4bfd17 /modules/sd_models.py | |
| parent | ce3f639ec8758ce2bc90483336361d2dc25acd3a (diff) | |
| download | stable-diffusion-webui-gfx803-0f8603a55988d22616b17140e6c4a7e9d0736af5.tar.gz stable-diffusion-webui-gfx803-0f8603a55988d22616b17140e6c4a7e9d0736af5.tar.bz2 stable-diffusion-webui-gfx803-0f8603a55988d22616b17140e6c4a7e9d0736af5.zip | |
add support for transformers==4.25.1
add fallback for when quick model creation fails
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 1bb9088b..b5bc12f0 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -14,7 +14,7 @@ import ldm.modules.midas as midas from ldm.util import instantiate_from_config
-from modules import shared, modelloader, devices, script_callbacks, sd_vae, sd_disable_initialization
+from modules import shared, modelloader, devices, script_callbacks, sd_vae, sd_disable_initialization, errors
from modules.paths import models_path
from modules.sd_hijack_inpainting import do_inpainting_hijack, should_hijack_inpainting
@@ -333,7 +333,11 @@ def load_model(checkpoint_info=None): timer = Timer()
- with sd_disable_initialization.DisableInitialization():
+ try:
+ with sd_disable_initialization.DisableInitialization():
+ sd_model = instantiate_from_config(sd_config.model)
+ except Exception as e:
+ print('Failed to create model quickly; will retry using slow method.', file=sys.stderr)
sd_model = instantiate_from_config(sd_config.model)
elapsed_create = timer.elapsed()
|