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