This commit is contained in:
nickwalton
2019-10-31 20:00:22 -06:00
parent 4ea1203084
commit ef78a64771
+1 -1
View File
@@ -221,7 +221,7 @@ class CTRLGenerator():
"http://www.@@", "edit@@", "*@@", "Writing", "Text@@", "\\@@", "<br>@@", "<div", "|@@", '...',
'..','', 'https://@@', '...@@']
encourage_tokens = ["apocalypse", "zombie", "radiation", "fallout", "undead", "corpse", "vampire", "virus", "plague"]
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