update

2026-09-20 12:20:53 +08:00 · 2019-09-26 16:39:51 -06:00
parent 0657dc6960
commit 1cd4d0c6f4
2 changed files with 4 additions and 4 deletions
@@ -217,7 +217,7 @@ class CTRLGenerator():

        # disallow some tokens
        forbidden_tokens = ['<unk>', 'Sco@@', "&amp@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@",
-                            "Edit", "&@@", "2:","1:", ":"]
+                            "Edit", "&@@", "2:","1:", ":", "Edit@@"]

        if num_new_lines > self.max_new_lines:
            forbidden_tokens.append("\n")