From 354ebba7892380d6935b9c9e0c72624e7fcced83 Mon Sep 17 00:00:00 2001 From: Anthony Wang Date: Fri, 15 Jul 2022 13:44:44 -0500 Subject: Use single-quotes instead of double quotes --- bot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bot.py b/bot.py index 9455406..f746e26 100644 --- a/bot.py +++ b/bot.py @@ -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) -- cgit v1.2.3-70-g09d2