Fix budget calculation for stories with at least one non-prompt chunk

This commit is contained in:
Gnome Ann
2021-11-03 18:22:31 -04:00
parent fe2987d894
commit 0fa47b1249

View File

@@ -1255,7 +1255,7 @@ def calcsubmit(txt):
tokens = memtokens + witokens + prompttkns + tokens tokens = memtokens + witokens + prompttkns + tokens
# Send completed bundle to generator # Send completed bundle to generator
ln = len(tokens) ln = len(tokens) + lnsp
if(not vars.model in ["Colab", "OAI"]): if(not vars.model in ["Colab", "OAI"]):
generate ( generate (