From e34d0edcec2f515e8a70ae059e3cb340e509582a Mon Sep 17 00:00:00 2001
From: nickwalton <nickwalton00@gmail.com>
Date: Thu, 31 Oct 2019 19:48:59 -0600
Subject: [PATCH] update

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

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 0c64180..6f8d934 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -221,6 +221,10 @@ class CTRLGenerator():
                             "http://www.@@", "edit@@", "*@@", "Writing", "Text@@", "\\@@", "<br>@@", "<div", "|@@", '...',
                             '..','…', 'https://@@', '...@@']
 
+        encourage_tokens = ["apocalypse", "zombies", "radiation", "fallout"]
+        for encourage_token in encourage_tokens:
+            prompt_logits[_token][self.word2idx[encourage_token]] *= 1.5
+
         for forbidden_token in forbidden_tokens:
             prompt_logits[_token][self.word2idx[forbidden_token]] = -1e8