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

fix the problem with infinite prompts where empty cond would be calculated incorrectly

Browse Source
This commit is contained in:
AUTOMATIC1111
2024-06-28 11:15:34 +03:00
parent 0b64633584
commit 179ae47d64
3 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modules/sd_models.py
View File

@@ -718,16 +718,15 @@ def get_empty_cond(sd_model):
p = processing.StableDiffusionProcessingTxt2Img()
extra_networks.activate(p, {})
if hasattr(sd_model, 'conditioner'):
if hasattr(sd_model, 'get_learned_conditioning'):
d = sd_model.get_learned_conditioning([""])
return d['crossattn']
else:
d = sd_model.cond_stage_model([""])
if isinstance(d, dict):
d = d['crossattn']
if isinstance(d, dict):
d = d['crossattn']
return d
return d
def send_model_to_cpu(m):