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

Ruff format

Browse Source
This commit is contained in:
Brandon Rising 2024-02-27 15:24:23 -05:00 committed by psychedelicious
parent f475b78734
commit c670dacc29
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -14,7 +14,10 @@ def extract_ti_triggers_from_prompt(prompt: str) -> List[str]:
ti_triggers.append(str(trigger)) ti_triggers.append(str(trigger))
return ti_triggers return ti_triggers
def generate_ti_list(prompt: str, base: BaseModelType, context: InvocationContext) -> List[Tuple[str, TextualInversionModelRaw]]:
def generate_ti_list(
prompt: str, base: BaseModelType, context: InvocationContext
) -> List[Tuple[str, TextualInversionModelRaw]]:
ti_list: List[Tuple[str, TextualInversionModelRaw]] = [] ti_list: List[Tuple[str, TextualInversionModelRaw]] = []
for trigger in extract_ti_triggers_from_prompt(prompt): for trigger in extract_ti_triggers_from_prompt(prompt):
name_or_key = trigger[1:-1] name_or_key = trigger[1:-1]