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 = ['', 'Sco@@', "UPDATE", "EDIT", "UPDATE", "EDIT", "[Part", "&"] + forbidden_tokens = ['', 'Sco@@', "EDIT", "[Part", "&"] if num_new_lines > self.max_new_lines: forbidden_tokens.append("\n")