Unable to train token classification model

Token Classification Notebook Link

—> 24 trainer.train()

6 frames
/usr/local/lib/python3.10/dist-packages/transformers/data/data_collator.py in torch_call(self, features)
308 no_labels_features = [{k: v for k, v in feature.items() if k != label_name} for feature in features]
309
→ 310 batch = self.tokenizer.pad(
311 no_labels_features,
312 padding=self.padding,

AttributeError: ‘DatasetDict’ object has no attribute ‘pad’

Any help will be highly appreciated.