Merge pull request #987 from huggingface/generative-finetuning

Generative finetuning
This commit is contained in:
Thomas Wolf authored and GitHub committed 2019-08-28 16:51:50 +02:00
commit 0ecfd17f49
11 files changed
+573 -6

No files matched your search

@@ -77,6 +77,9 @@ class RobertaTokenizer(PreTrainedTokenizer):
sep_token=sep_token, cls_token=cls_token, pad_token=pad_token,
mask_token=mask_token, **kwargs)
self.max_len_single_sentence = self.max_len - 2 # take into account special tokens
self.max_len_sentences_pair = self.max_len - 4 # take into account special tokens
self.encoder = json.load(open(vocab_file, encoding="utf-8"))
self.decoder = {v: k for k, v in self.encoder.items()}
self.errors = errors # how to handle errors in decoding