Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "comment out customer_attention_context"

Browse Source 
This reverts commit 8f8cd90787.

Due to NameError: name 'options' is not defined
This commit is contained in:
Eugene 2023-05-14 00:37:55 -04:00
parent 2623941d91
commit b72c9787a9
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
invokeai/backend/prompting/conditioning.py
View File

@ -59,15 +59,12 @@ def get_uc_and_c_and_ec(prompt_string,
if log_tokens or getattr(Globals, "log_tokenization", False):
log_tokenization(positive_prompt, negative_prompt, tokenizer=model.tokenizer)
# The below has been commented out as it is an instance method used for cleanly loading LoRA models, but is not currently needed.
# TODO: Reimplement custom_attention for 3.0 support of LoRA.
# with InvokeAIDiffuserComponent.custom_attention_context(model.unet,
# extra_conditioning_info=None,
# step_count=-1):
# c, options = compel.build_conditioning_tensor_for_prompt_object(positive_prompt)
# uc, _ = compel.build_conditioning_tensor_for_prompt_object(negative_prompt)
# [c, uc] = compel.pad_conditioning_tensors_to_same_length([c, uc])
with InvokeAIDiffuserComponent.custom_attention_context(model.unet,
extra_conditioning_info=None,
step_count=-1):
c, options = compel.build_conditioning_tensor_for_prompt_object(positive_prompt)
uc, _ = compel.build_conditioning_tensor_for_prompt_object(negative_prompt)
[c, uc] = compel.pad_conditioning_tensors_to_same_length([c, uc])
# now build the "real" ec
ec = InvokeAIDiffuserComponent.ExtraConditioningInfo(tokens_count_including_eos_bos=tokens_count,