Hotfix for Google Colab generator call failing if when called from a fresh prompt.

2021-05-13 23:30:54 -04:00 · 2021-05-13 23:30:54 -04:00 · c9b6f89d1d
parent 3c0638bc73
commit c9b6f89d1d
1 changed files with 4 additions and 1 deletions
--- 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     = []