Add an assertion for the value of already_generated

This commit is contained in:
Gnome Ann 2021-11-03 12:01:45 -04:00
parent ec8ec55256
commit de3664e73c
1 changed files with 1 additions and 0 deletions

View File

@ -1440,6 +1440,7 @@ 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)