From 7a5223f203a187d7a8cb28f9396aedeadcb98dad Mon Sep 17 00:00:00 2001 From: nickwalton Date: Thu, 31 Oct 2019 20:04:01 -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 98355d9..eda359e 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -223,7 +223,7 @@ class CTRLGenerator(): encourage_tokens = ["zombie", "radiation", "fallout", "undead", "corpse", "vampire", "virus", "plague"] for encourage_token in encourage_tokens: - prompt_logits[_token][self.word2idx[encourage_token]] *= 1.5 + prompt_logits[_token][self.word2idx[encourage_token]] *= 1.2 for forbidden_token in forbidden_tokens: prompt_logits[_token][self.word2idx[forbidden_token]] = -1e8