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

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

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