From aa07103aeec94b5ef03e1c9f71aa952032cef3f6 Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Thu, 26 Sep 2019 12:05:41 -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 0f1f674..87cae33 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@@', "[Part", "&amp"]
+        forbidden_tokens = ['<unk>', 'Sco@@']
 
         if num_new_lines > self.max_new_lines:
             forbidden_tokens.append("\n")