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

Text in AI highlighting fix

Browse Source
This commit is contained in:
ebolam
2022-09-20 09:56:52 -04:00
parent 00dd0d1c9b
commit 96ab15c2e3
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
koboldai_settings.py
View File

@@ -154,7 +154,8 @@ class koboldai_vars(object):
#Add prompt lenght/text if we're set to always use prompt
if self.useprompt:
if self.max_prompt_length if self.prompt_length > self.max_prompt_length else self.prompt_length + used_tokens < token_budget:
self.max_prompt_length if self.prompt_length > self.max_prompt_length else self.prompt_length
if prompt_length + used_tokens < token_budget:
used_tokens += self.max_prompt_length if self.prompt_length > self.max_prompt_length else self.prompt_length
#Find World Info entries in prompt
for wi in self.worldinfo_v2:
@@ -241,7 +242,8 @@ class koboldai_vars(object):
game_context.insert(0, {"type": "authors_note", "text": authors_note_final})
if not self.useprompt:
if self.max_prompt_length if self.prompt_length > self.max_prompt_length else self.prompt_length + used_tokens < token_budget:
prompt_length = self.max_prompt_length if self.prompt_length > self.max_prompt_length else self.prompt_length
if prompt_length + used_tokens < token_budget:
used_tokens += self.max_prompt_length if self.prompt_length > self.max_prompt_length else self.prompt_length
#Find World Info entries in prompt
for wi in self.worldinfo_v2: