From d134503c942be35f354744ce55f38e795ba9b332 Mon Sep 17 00:00:00 2001 From: Nick Date: Fri, 20 Sep 2019 13:25:13 -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 dbbc050..a256846 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -212,7 +212,7 @@ class CTRLGenerator(): # Make sure only a possible verb is chosen. if first_token: for word in get_possible_verbs(): - prompt_logits[_token][self.word2idx[word]] += max(prompt_logits) + prompt_logits[_token][self.word2idx[word]] += 10 # compute probabilities from logits prompt_probs = np.exp(prompt_logits[_token])