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

fix(nodes): fix TI loading

Browse Source
This commit is contained in:
psychedelicious 2024-02-21 19:04:43 +11:00
parent e3c23baae9
commit 34f3a39cc9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -86,7 +86,7 @@ class CompelInvocation(BaseInvocation):
for trigger in extract_ti_triggers_from_prompt(self.prompt): for trigger in extract_ti_triggers_from_prompt(self.prompt):
name = trigger[1:-1] name = trigger[1:-1]
try: try:
loaded_model = context.models.load(**self.clip.text_encoder.model_dump()).model loaded_model = context.models.load(key=name).model
assert isinstance(loaded_model, TextualInversionModelRaw) assert isinstance(loaded_model, TextualInversionModelRaw)
ti_list.append((name, loaded_model)) ti_list.append((name, loaded_model))
except UnknownModelException: except UnknownModelException: