Merge pull request #66 from Aktsvigun/master

HHousen · Feb 3, 2022 · 176d394 · 176d394
2 parents 8165890 + 73aef8d
commit 176d394
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/abstractive.py b/src/abstractive.py
@@ -142,6 +142,7 @@ def __init__(self, hparams):
             self.target_eoseq_token = self.tokenizer.pad_token
         else:
             self.target_eoseq_token = "[unused1]"
+            self.tokenizer.pad_token = "[unused2]"
             do_seq_special_add = True
 
         # Convert `target_boseq_token` and `target_eoseq_token` to IDs