Index Error when fine tune the pretrained model

[richardcuick]

Index Error when fine tune the pretrained model with following notebook, do you know how to resolve the problem?
https://github.com/keitazoumana/Medium-Articles-Notebooks/blob/main/Train_your_LLM.ipynb