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

update

Browse Source
This commit is contained in:
Nick
2019-09-19 23:08:15 -06:00
parent a73939691a
commit 44e1b7860a
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/ctrl/ctrl_generator.py
+1
View File
@@ -199,6 +199,7 @@ class CTRLGenerator():
# disallow some tokens
prompt_logits[_token][self.word2idx['<unk>']] = -1e8
prompt_logits[_token][self.word2idx['\n']] = -1e8
# sometimes, when generating from reddit,
# it tries to generate the Score (reddit Karma) immediately after generating the Title: