Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-04-30 19:21:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Use [PROMPT] keyword to insert the source prompt (#243)

Browse Source
This commit is contained in:
Danamir
2023-08-08 13:13:57 +02:00
committed by GitHub
parent f18e00b408
commit 3e80b2d824
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/!adetailer.py
View File

@@ -244,6 +244,8 @@ class AfterDetailerScript(scripts.Script):
for n in range(len(prompts)): for n in range(len(prompts)):
if not prompts[n]: if not prompts[n]:
prompts[n] = blank_replacement prompts[n] = blank_replacement
elif "[PROMPT]" in prompts[n]:
prompts[n] = prompts[n].replace("[PROMPT]", f" {blank_replacement} ")
return prompts return prompts
def get_prompt(self, p, args: ADetailerArgs) -> tuple[list[str], list[str]]: def get_prompt(self, p, args: ADetailerArgs) -> tuple[list[str], list[str]]: