wassname/Clover-Edition
1
0
Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-09-09 11:13:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
nickwalton
2019-10-31 20:04:01 -06:00
parent ef78a64771
commit 7a5223f203
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/ctrl/ctrl_generator.py
+1 -1
View File
@@ -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