Skip to content

Commit 877f76e

Browse files
committed
add small nit
1 parent 609b880 commit 877f76e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

evaluation/generation/generate.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,9 @@ def main():
4848
text = ''
4949
while True:
5050
try:
51-
dummy = input('''Enter the paragraph (Enter for new line and Ctrl-c to end the prompt):''')+'\n'
51+
dummy = input('''Enter the paragraph (Enter for new line and Ctrl-c to end the prompt):''')
5252
text += dummy
53+
text += '\n'
5354
except KeyboardInterrupt:
5455
output = generate_from_text(model, text, tokenizer, max_length=args.generate_max_length, greedy=args.greedy, top_k=args.top_k)
5556
print(output)

0 commit comments

Comments
 (0)