rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Update author's template effect token usage live

Browse Source
This commit is contained in:
somebody
2022-08-11 18:38:29 -05:00
parent c21c1e3dc0
commit 6ac970b1c0
2 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
aiserver.py
View File

@ -3634,7 +3634,7 @@ def get_message(msg):
)
else:
if field == "anoteinput":
unencoded = buildauthorsnote(unencoded)
unencoded = buildauthorsnote(unencoded, msg["data"]["anotetemplate"])
tokens_length = len(tokenizer.encode(unencoded))
emit(
@ -3972,11 +3972,11 @@ def actionredo():
#==================================================================#
#
#==================================================================#
def buildauthorsnote(authorsnote):
def buildauthorsnote(authorsnote, template):
# Build Author's Note if set
if authorsnote == "":
return ""
return ("\n" + vars.authornotetemplate + "\n").replace("<|>", authorsnote)
return ("\n" + template + "\n").replace("<|>", authorsnote)
def calcsubmitbudgetheader(txt, **kwargs):
# Scan for WorldInfo matches
@ -3988,7 +3988,7 @@ def calcsubmitbudgetheader(txt, **kwargs):
else:
mem = vars.memory
anotetxt = buildauthorsnote(vars.authornote)
anotetxt = buildauthorsnote(vars.authornote, vars.authornotetemplate)
return winfo, mem, anotetxt, found_entries