From 4c2e4532d65a87e2f68f063a8483aaa9c228eab8 Mon Sep 17 00:00:00 2001
From: Nick <nickwalton00@gmail.com>
Date: Tue, 29 Oct 2019 08:58:36 -0600
Subject: [PATCH] update

---
 generator/ctrl/generation.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/generator/ctrl/generation.py b/generator/ctrl/generation.py
index b029926..61f253d 100644
--- a/generator/ctrl/generation.py
+++ b/generator/ctrl/generation.py
@@ -212,6 +212,7 @@ while True:
 
           # disallow some tokens
           prompt_logits[_token][word2idx['<unk>']] = -1e8
+          prompt_logits[_token][word2idx['\n']] = -1e8
 
           # sometimes, when generating from reddit,
           # it tries to generate the Score (reddit Karma) immediately after generating the Title: