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

Update generate.py

Browse Source
This commit is contained in:
Jordan
2023-02-20 21:33:20 -07:00
parent 8f527c2b2d
commit e2b6dfeeb9
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ldm/generate.py
View File

@@ -482,6 +482,8 @@ class Generate:
if self.model.lora_manager: if self.model.lora_manager:
prompt = self.model.lora_manager.configure_prompt(prompt) prompt = self.model.lora_manager.configure_prompt(prompt)
# lora MUST process prompt before conditioning
self.model.lora_manager.load_lora()
# apply the concepts library to the prompt # apply the concepts library to the prompt
prompt = self.huggingface_concepts_library.replace_concepts_with_triggers( prompt = self.huggingface_concepts_library.replace_concepts_with_triggers(
@@ -513,10 +515,6 @@ class Generate:
except AttributeError: except AttributeError:
pass pass
# lora MUST process prompt before conditioning
if self.model.lora_manager:
self.model.lora_manager.load_lora()
try: try:
uc, c, extra_conditioning_info = get_uc_and_c_and_ec( uc, c, extra_conditioning_info = get_uc_and_c_and_ec(
prompt, prompt,