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

2021-11-03 18:22:31 -04:00 · 2021-11-03 18:22:31 -04:00 · 0fa47b1249
parent fe2987d894
commit 0fa47b1249
1 changed files with 1 additions and 1 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -1255,7 +1255,7 @@ def calcsubmit(txt):
                tokens = memtokens + witokens + prompttkns + tokens
            
            # Send completed bundle to generator
-            ln = len(tokens)
+            ln = len(tokens) + lnsp
            
            if(not vars.model in ["Colab", "OAI"]):
                generate (