From c11dab894ebc7ce281c4d750355beb493fe2c121 Mon Sep 17 00:00:00 2001
From: Gnome Ann <>
Date: Wed, 3 Nov 2021 18:02:19 -0400
Subject: [PATCH] Put placeholder variables into calcsubmitbudget

---
 aiserver.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/aiserver.py b/aiserver.py
index e0a3a466..a74c68e9 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -1224,6 +1224,8 @@ def calcsubmitbudgetheader(txt, **kwargs):
     return winfo, mem, anotetxt, found_entries
 
 def calcsubmitbudget(actionlen, winfo, mem, anotetxt, actions=vars.actions):
+    forceanote   = False # In case we don't have enough actions to hit A.N. depth
+    anoteadded   = False # In case our budget runs out before we hit A.N. depth
     anotetkns    = []  # Placeholder for Author's Note tokens
     lnanote      = 0   # Placeholder for Author's Note length
 
@@ -1457,7 +1459,6 @@ def generate(txt, min, max):
                     )
                 already_generated += len(genout[0]) - len(gen_in[0])
                 if(not model.kai_scanner.any_new_entries):
-                    assert already_generated == max - min + 1
                     break
                 txt = tokenizer.decode(genout[0, -already_generated:])
                 winfo, mem, anotetxt, _found_entries = calcsubmitbudgetheader(txt, force_use_txt=True)