This commit is contained in:
Lysandre committed 2019-10-22 14:12:33 -04:00
1 parent 44286b94d3
commit 7d709e55ed
10 files changed
+41 -39

No files matched your search

@@ -68,7 +68,7 @@ def main():
start = time.time()
for text in data:
text = f'{bos} {text.strip()} {sep}'
token_ids = tokenizer.encode(text)
token_ids = tokenizer.encode(text, add_special_tokens=False)
rslt.append(token_ids)
iter += 1