From e67f192cc3cd7f535362af6bc3d3f9beafb8a1c9 Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Thu, 26 Sep 2019 12:02:17 -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 01fd396..b0894ee 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -216,7 +216,7 @@ class CTRLGenerator():
                 prompt_logits[_token][generated_token] /= self.penalty
 
         # disallow some tokens
-        forbidden_tokens = ['<unk>', 'Sco@@', "UPDATE", "EDIT", "UPDATE", "EDIT", "[Part", "&amp"]
+        forbidden_tokens = ['<unk>', 'Sco@@', "EDIT", "[Part", "&amp"]
 
         if num_new_lines > self.max_new_lines:
             forbidden_tokens.append("\n")