github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

prevent crash when prompt blend requested

Browse Source
This commit is contained in:
Lincoln Stein
2023-04-06 21:22:47 -04:00
parent 0784e49d92
commit 35c4ff8ab0
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
ldm/invoke/conditioning.py
View File

@@ -63,12 +63,13 @@ def get_uc_and_c_and_ec(prompt_string, model, log_tokens=False, skip_normalize_l
else:
positive_prompt = Compel.parse_prompt_string(positive_prompt_string)
should_use_lora_manager = True
lora_weights = getattr(positive_prompt, 'lora_weights', [])
if model.peft_manager:
should_use_lora_manager = model.peft_manager.should_use(positive_prompt.lora_weights)
should_use_lora_manager = model.peft_manager.should_use(lora_weights)
if not should_use_lora_manager:
model.peft_manager.set_loras(positive_prompt.lora_weights)
model.peft_manager.set_loras(lora_weights)
if model.lora_manager and should_use_lora_manager:
lora_conditions = model.lora_manager.set_loras_conditions(positive_prompt.lora_weights)
lora_conditions = model.lora_manager.set_loras_conditions(lora_weights)
negative_prompt: FlattenedPrompt|Blend = Compel.parse_prompt_string(negative_prompt_string)
if log_tokens or getattr(Globals, "log_tokenization", False):