fix model loading twice in some situations
This commit is contained in:
parent
5f5435eb1a
commit
1a43524018
|
@ -216,6 +216,9 @@ class StableDiffusionModelHijack:
|
|||
self.comments = []
|
||||
|
||||
def get_prompt_lengths(self, text):
|
||||
if self.clip is None:
|
||||
return "-", "-"
|
||||
|
||||
_, token_count = self.clip.process_texts([text])
|
||||
|
||||
return token_count, self.clip.get_target_prompt_token_count(token_count)
|
||||
|
|
|
@ -414,6 +414,9 @@ class SdModelData:
|
|||
def get_sd_model(self):
|
||||
if self.sd_model is None:
|
||||
with self.lock:
|
||||
if self.sd_model is not None:
|
||||
return self.sd_model
|
||||
|
||||
try:
|
||||
load_model()
|
||||
except Exception as e:
|
||||
|
|
Loading…
Reference in New Issue