diff --git a/generator/ctrl/generation.py b/generator/ctrl/generation.py index b029926..61f253d 100644 --- a/generator/ctrl/generation.py +++ b/generator/ctrl/generation.py @@ -212,6 +212,7 @@ while True: # disallow some tokens prompt_logits[_token][word2idx['']] = -1e8 + prompt_logits[_token][word2idx['\n']] = -1e8 # sometimes, when generating from reddit, # it tries to generate the Score (reddit Karma) immediately after generating the Title: