fixed compel.py as requested

This commit is contained in:
Lincoln Stein 2023-05-10 21:40:44 -04:00
parent 5d5157fc65
commit aca4770481
2 changed files with 3 additions and 2 deletions

View File

@ -100,7 +100,8 @@ class CompelInvocation(BaseInvocation):
# TODO: support legacy blend?
prompt: Union[FlattenedPrompt, Blend] = Compel.parse_prompt_string(prompt_str)
conjunction = Compel.parse_prompt_string(prompt_str)
prompt: Union[FlattenedPrompt, Blend] = conjunction.prompts[0]
if getattr(Globals, "log_tokenization", False):
log_tokenization_for_prompt_object(prompt, tokenizer)