mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

oh hey i broke batches with this

Browse Source
This commit is contained in:
AUTOMATIC 2022-09-15 18:05:42 +03:00
parent cde49dcbe6
commit 0fde1f3eac
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/prompt_parser.py
View File

@ -34,6 +34,7 @@ def get_learned_conditioning_prompt_schedules(prompts, steps):
cached = cache.get(prompt, None) cached = cache.get(prompt, None)
if cached is not None: if cached is not None:
res.append(cached) res.append(cached)
continue
for m in re_prompt.finditer(prompt): for m in re_prompt.finditer(prompt):
plaintext = m.group(1) if m.group(5) is None else m.group(5) plaintext = m.group(1) if m.group(5) is None else m.group(5)
@ -97,6 +98,7 @@ def get_learned_conditioning(prompts, steps):
cached = cache.get(prompt, None) cached = cache.get(prompt, None)
if cached is not None: if cached is not None:
res.append(cached) res.append(cached)
continue
texts = [x[1] for x in prompt_schedule] texts = [x[1] for x in prompt_schedule]
conds = shared.sd_model.get_learned_conditioning(texts) conds = shared.sd_model.get_learned_conditioning(texts)