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

This commit is contained in:
KoboldAI Dev 2021-05-13 23:30:54 -04:00
parent 3c0638bc73
commit c9b6f89d1d
1 changed files with 4 additions and 1 deletions

View File

@ -574,7 +574,10 @@ def calcsubmit(txt):
subtxt = vars.memory + winfo + anotetxt + vars.prompt subtxt = vars.memory + winfo + anotetxt + vars.prompt
lnsub = lnmem + lnwi + lnprompt + lnanote lnsub = lnmem + lnwi + lnprompt + lnanote
if(vars.model != "Colab"):
generate(subtxt, lnsub+1, lnsub+vars.genamt) generate(subtxt, lnsub+1, lnsub+vars.genamt)
else:
sendtocolab(subtxt, lnsub+1, lnsub+vars.genamt)
else: else:
tokens = [] tokens = []