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)

View File

@ -38,7 +38,7 @@ dependencies = [
"albumentations",
"click",
"clip_anytorch", # replacing "clip @ https://github.com/openai/CLIP/archive/eaa22acb90a5876642d0507623e859909230a52d.zip",
"compel~=1.1.5",
"compel~=1.1.5",
"datasets",
"diffusers[torch]~=0.16.1",
"dnspython==2.2.1",