Skip to content

Commit 560b3fc

Browse files
committed
fix small nit
1 parent 877f76e commit 560b3fc

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

evaluation/generation/generate.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,8 @@ 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):''')
51+
dummy = input('''Enter the paragraph (Enter for new line and Ctrl-c to end the prompt):''')+'\n'
5252
text += dummy
53-
text += '\n'
5453
except KeyboardInterrupt:
5554
output = generate_from_text(model, text, tokenizer, max_length=args.generate_max_length, greedy=args.greedy, top_k=args.top_k)
5655
print(output)

0 commit comments

Comments
 (0)