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

Next: Switch SDXLPromptInvocationBase to read TI names as model keys rather than model_name

Browse Source
This commit is contained in:
Brandon Rising 2024-02-27 10:15:39 -05:00 committed by Brandon
parent 4418c118db
commit 33856def7c
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/invocations/compel.py
View File

@ -193,11 +193,9 @@ class SDXLPromptInvocationBase:
for trigger in extract_ti_triggers_from_prompt(prompt): for trigger in extract_ti_triggers_from_prompt(prompt):
name = trigger[1:-1] name = trigger[1:-1]
try: try:
ti_model = context.models.load_by_attrs( loaded_model = context.models.load(key=name).model
model_name=name, base_model=text_encoder_info.config.base, model_type=ModelType.TextualInversion assert isinstance(loaded_model, TextualInversionModelRaw)
).model ti_list.append((name, loaded_model))
assert isinstance(ti_model, TextualInversionModelRaw)
ti_list.append((name, ti_model))
except UnknownModelException: except UnknownModelException:
# print(e) # print(e)
# import traceback # import traceback

8
invokeai/app/util/ti_utils.py
View File

@ -1,8 +1,10 @@
import re import re
from typing import List
def extract_ti_triggers_from_prompt(prompt: str) -> list[str]:
ti_triggers = [] def extract_ti_triggers_from_prompt(prompt: str) -> List[str]:
ti_triggers: List[str] = []
for trigger in re.findall(r"<[a-zA-Z0-9., _-]+>", prompt): for trigger in re.findall(r"<[a-zA-Z0-9., _-]+>", prompt):
ti_triggers.append(trigger) ti_triggers.append(str(trigger))
return ti_triggers return ti_triggers