From 44e1b7860a1b3e87ef884445317de98075a91a4a Mon Sep 17 00:00:00 2001
From: Nick <nickwalton00@gmail.com>
Date: Thu, 19 Sep 2019 23:08:15 -0600
Subject: [PATCH] update

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

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 6066e8d..d620de7 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -199,6 +199,7 @@ class CTRLGenerator():
 
             # disallow some tokens
             prompt_logits[_token][self.word2idx['<unk>']] = -1e8
+            prompt_logits[_token][self.word2idx['\n']] = -1e8
 
             # sometimes, when generating from reddit,
             # it tries to generate the Score (reddit Karma) immediately after generating the Title: