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

Use single-quotes instead of double quotes

Browse source
This commit is contained in:
Anthony Wang 2022-07-15 13:44:44 -05:00
parent fc56d2c0a8
commit 354ebba789
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

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