StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

if sd_model is None, do not always try to load it

Browse Source
This commit is contained in:
AUTOMATIC 2023-05-18 15:47:43 +03:00
parent ae252cd5bc
commit cd8a510ca9
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
modules/sd_models.py
View File

@ -409,12 +409,16 @@ sd2_clip_weight = 'cond_stage_model.model.transformer.resblocks.0.attn.in_proj_w
class SdModelData: class SdModelData:
def __init__(self): def __init__(self):
self.sd_model = None self.sd_model = None
self.was_loaded_at_least_once = False
self.lock = threading.Lock() self.lock = threading.Lock()
def get_sd_model(self): def get_sd_model(self):
if self.was_loaded_at_least_once:
return self.sd_model
if self.sd_model is None: if self.sd_model is None:
with self.lock: with self.lock:
if self.sd_model is not None: if self.sd_model is not None or self.was_loaded_at_least_once:
return self.sd_model return self.sd_model
try: try:
@ -495,6 +499,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
sd_model.eval() sd_model.eval()
model_data.sd_model = sd_model model_data.sd_model = sd_model
model_data.was_loaded_at_least_once = True
sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings(force_reload=True) # Reload embeddings after model load as they may or may not fit the model sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings(force_reload=True) # Reload embeddings after model load as they may or may not fit the model