From b97de0808c534930c39ba04164c19991fbcbd897 Mon Sep 17 00:00:00 2001
From: Nick <nickwalton00@gmail.com>
Date: Wed, 30 Oct 2019 10:58:16 -0600
Subject: [PATCH] update

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

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index b5ef26e..60273a6 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -227,6 +227,8 @@ class CTRLGenerator():
 
         if forbid_newline:
             prompt_logits[_token][self.word2idx['\n']] = -1e8
+        else:
+            prompt_logits[_token][self.word2idx['\n']] *= 0.75
 
         # Set whitelist
         if "word_whitelist" in options and token_num in options["word_whitelist"].keys():