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 1/6] 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
 

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 2/6] 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
 

From 4ea12030842d68cbd089f6e0aa5b0257074675d5 Mon Sep 17 00:00:00 2001
From: nickwalton <nickwalton00@gmail.com>
Date: Thu, 31 Oct 2019 19:58:38 -0600
Subject: [PATCH 3/6] update

---
 generator/ctrl/ctrl_generator.py | 2 +-
 story/utils.py                   | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index a41d968..42b1d5b 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", "undead"]
+        encourage_tokens = ["apocalypse", "zombie", "radiation", "fallout", "undead", "corpse", "vampire", "virus", "plague"]
         for encourage_token in encourage_tokens:
             prompt_logits[_token][self.word2idx[encourage_token]] *= 1.5
 
diff --git a/story/utils.py b/story/utils.py
index a03cadf..1a78b09 100644
--- a/story/utils.py
+++ b/story/utils.py
@@ -119,7 +119,11 @@ first_to_second_mappings = [
     ("Im", "you're"),
     ("Ive", "you've"),
     ("I am", "you are"),
+    ("was I", "were you"),
+    ("am I", "are you"),
+    ("wasn't I", "weren't you"),
     ("I", "you"),
+    ("I'd", "you'd"),
     ("i", "you"),
     ("I've", "you've"),
     ("my", "your"),

From ef78a6477199c40d86308880e992137fbc5cf7e6 Mon Sep 17 00:00:00 2001
From: nickwalton <nickwalton00@gmail.com>
Date: Thu, 31 Oct 2019 20:00:22 -0600
Subject: [PATCH 4/6] 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 42b1d5b..98355d9 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", "zombie", "radiation", "fallout", "undead", "corpse", "vampire", "virus", "plague"]
+        encourage_tokens = ["zombie", "radiation", "fallout", "undead", "corpse", "vampire", "virus", "plague"]
         for encourage_token in encourage_tokens:
             prompt_logits[_token][self.word2idx[encourage_token]] *= 1.5
 

From 7a5223f203a187d7a8cb28f9396aedeadcb98dad Mon Sep 17 00:00:00 2001
From: nickwalton <nickwalton00@gmail.com>
Date: Thu, 31 Oct 2019 20:04:01 -0600
Subject: [PATCH 5/6] 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 98355d9..eda359e 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -223,7 +223,7 @@ class CTRLGenerator():
 
         encourage_tokens = ["zombie", "radiation", "fallout", "undead", "corpse", "vampire", "virus", "plague"]
         for encourage_token in encourage_tokens:
-            prompt_logits[_token][self.word2idx[encourage_token]] *= 1.5
+            prompt_logits[_token][self.word2idx[encourage_token]] *= 1.2
 
         for forbidden_token in forbidden_tokens:
             prompt_logits[_token][self.word2idx[forbidden_token]] = -1e8

From 93133cfce7068261e646d47faa0c84119108be4b Mon Sep 17 00:00:00 2001
From: nickwalton <nickwalton00@gmail.com>
Date: Thu, 31 Oct 2019 20:33:21 -0600
Subject: [PATCH 6/6] update

---
 story/utils.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/story/utils.py b/story/utils.py
index 1a78b09..83ddbd5 100644
--- a/story/utils.py
+++ b/story/utils.py
@@ -126,6 +126,7 @@ first_to_second_mappings = [
     ("I'd", "you'd"),
     ("i", "you"),
     ("I've", "you've"),
+    ("I was", "you were"),
     ("my", "your"),
     ("we","you"),
     ("we're", "you're"),
@@ -141,6 +142,8 @@ second_to_first_mappings = [
     ("you're", "I'm"),
     ("your", "my"),
     ("you are", "I am"),
+    ("you were", "I was"),
+    ("are you", "am I"),
     ("you", "I"),
     ("you", "me"),
     ("you'll", "I'll"),