Error if prompt missing SR token to prevent mis-gens (#2209)

This commit is contained in:
aperullo 2022-10-11 06:16:57 -04:00 committed by GitHub
parent 948533950c
commit 255be75d30
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions

View File

@ -27,9 +27,16 @@ def apply_field(field):
def apply_prompt(p, x, xs):
orig_prompt = p.prompt
orig_negative_prompt = p.negative_prompt
p.prompt = p.prompt.replace(xs[0], x)
p.negative_prompt = p.negative_prompt.replace(xs[0], x)
if p.prompt == orig_prompt and p.negative_prompt == orig_negative_prompt:
raise RuntimeError(f"Prompt S/R did not find {xs[0]} in prompt or negative prompt. Did you forget to add the token?")
def apply_order(p, x, xs):
token_order = []