Fixed BPE.read_files -> BPE.read_file in SentencePieceBPETokenizer

This commit is contained in:
Timur Ganiev
2020-10-23 17:30:38 +03:00
committed by Anthony MOI
parent a2289d49b4
commit f7c61c267a

View File

@@ -49,7 +49,7 @@ class SentencePieceBPETokenizer(BaseTokenizer):
@staticmethod @staticmethod
def from_file(vocab_filename: str, merges_filename: str, **kwargs): def from_file(vocab_filename: str, merges_filename: str, **kwargs):
vocab, merges = BPE.read_files(vocab_filename, merges_filename) vocab, merges = BPE.read_file(vocab_filename, merges_filename)
return SentencePieceBPETokenizer(vocab, merges, **kwargs) return SentencePieceBPETokenizer(vocab, merges, **kwargs)
def train( def train(