Added the necessay import of module (#30804)

2024-05-14 23:15:06 +05:30
parent ccdabc5642
commit c96aca3a8d
1 changed files with 1 additions and 0 deletions
--- 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)