hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add support for transformers==4.25.1

Browse Source 
add fallback for when quick model creation fails
This commit is contained in:
AUTOMATIC
2023-01-10 17:46:59 +03:00
parent ce3f639ec8
commit 0f8603a559
2 changed files with 42 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules/sd_models.py
View File

@@ -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()