From c96aca3a8d66d64f868a3e3967be624d79213bef Mon Sep 17 00:00:00 2001 From: Ankur Singh Date: Tue, 14 May 2024 23:15:06 +0530 Subject: [PATCH] Added the necessay import of module (#30804) --- docs/source/en/training.md | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/source/en/training.md b/docs/source/en/training.md index cea583c05e..aacf174fbd 100644 --- a/docs/source/en/training.md +++ b/docs/source/en/training.md @@ -186,6 +186,7 @@ so we can just convert that directly to a NumPy array without tokenization! ```py from transformers import AutoTokenizer +import numpy as np tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-cased") tokenized_data = tokenizer(dataset["sentence"], return_tensors="np", padding=True)