From c9b6f89d1d6fe6b47b094f64a3ce418804c82831 Mon Sep 17 00:00:00 2001
From: KoboldAI Dev <KoboldAIDev@gmail.com>
Date: Thu, 13 May 2021 23:30:54 -0400
Subject: [PATCH] Hotfix for Google Colab generator call failing if when called
 from a fresh prompt.

---
 aiserver.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/aiserver.py b/aiserver.py
index 28cd3ce5..0e3d33d3 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -574,7 +574,10 @@ def calcsubmit(txt):
             subtxt = vars.memory + winfo + anotetxt + vars.prompt
             lnsub  = lnmem + lnwi + lnprompt + lnanote
             
-            generate(subtxt, lnsub+1, lnsub+vars.genamt)
+            if(vars.model != "Colab"):
+                generate(subtxt, lnsub+1, lnsub+vars.genamt)
+            else:
+                sendtocolab(subtxt, lnsub+1, lnsub+vars.genamt)
         else:
             tokens     = []