This commit is contained in:
Nick
2019-09-26 19:59:57 -06:00
parent bdd888282b
commit eaee91a93a
+1 -1
View File
@@ -217,7 +217,7 @@ class CTRLGenerator():
# disallow some tokens
forbidden_tokens = ['<unk>', 'Sco@@', "&amp@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@",
"Edit", "&@@", "2:","1:", ":", "Edit@@", "EDI@@", "EDIT@@"]
"Edit", "&@@", "2:","1:", ":", "Edit@@", "EDI@@", "EDIT@@", "edit", "TL@@"]
if num_new_lines > self.max_new_lines:
forbidden_tokens.append("\n")