OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-01-01 12:25:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix model loading twice in some situations

Browse Source
This commit is contained in:
AUTOMATIC 2023-05-14 13:27:50 +03:00
parent 5f5435eb1a
commit 1a43524018
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/sd_hijack.py
View File

@ -216,6 +216,9 @@ class StableDiffusionModelHijack:
self.comments = [] self.comments = []
def get_prompt_lengths(self, text): def get_prompt_lengths(self, text):
if self.clip is None:
return "-", "-"
_, token_count = self.clip.process_texts([text]) _, token_count = self.clip.process_texts([text])
return token_count, self.clip.get_target_prompt_token_count(token_count) return token_count, self.clip.get_target_prompt_token_count(token_count)

3
modules/sd_models.py
View File

@ -414,6 +414,9 @@ class SdModelData:
def get_sd_model(self): def get_sd_model(self):
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:
return self.sd_model
try: try:
load_model() load_model()
except Exception as e: except Exception as e: