rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Fix budget calculation when using soft prompt

This commit is contained in:
Gnome Ann 2021-10-29 11:44:51 -04:00
parent bf4e7742ac
commit 206c01008e
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
aiserver.py
View File

@ -1190,9 +1190,6 @@ def calcsubmit(txt):
else: else:
budget = vars.max_length - lnsp - lnmem - lnanote - lnwi - vars.genamt budget = vars.max_length - lnsp - lnmem - lnanote - lnwi - vars.genamt
if(vars.sp is not None):
budget -= vars.sp.shape[0]
if(actionlen == 0): if(actionlen == 0):
# First/Prompt action # First/Prompt action
subtxt = vars.memory + winfo + anotetxt + vars.prompt subtxt = vars.memory + winfo + anotetxt + vars.prompt