a/ebooks
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Increase maximum word count

Browse source
This commit is contained in:
Anthony Wang 2022-02-22 22:44:17 -06:00
parent 758998d17e
commit 3010b24685
Signed by: a
GPG key ID: BC96B00AEC5F2D76
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
bot.py
View file

@ -54,7 +54,7 @@ if args.input is None:
print(args.input)
inputs = tokenizer.encode(args.input, return_tensors="pt")
output = tokenizer.decode(model.generate(
inputs, do_sample=True, max_length=100, top_p=0.9)[0])
inputs, do_sample=True, max_length=150, top_p=0.9)[0])
print(output)