update

2026-09-11 11:51:53 +08:00 · 2019-09-20 13:25:13 -06:00
parent 80bd0ec4f0
commit d134503c94
1 changed files with 1 additions and 1 deletions
@@ -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])