From 0680a77442f88d13c063e20b051d2c0f5ac297d6 Mon Sep 17 00:00:00 2001 From: Nick Date: Fri, 20 Sep 2019 12:57:25 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 33f6881..3de5673 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -201,8 +201,8 @@ class CTRLGenerator(): # disallow some tokens forbidden_tokens = ['', 'Sco@@', '\n'] - for token in forbidden_tokens: - prompt_logits[_token][self.word2idx[token]] = -1e8 + for forbidden_token in forbidden_tokens: + prompt_logits[_token][self.word2idx[forbidden_token]] = -1e8 # Make sure only a possible verb is chosen. if first_token: