I am currently trying to run a helper function to train a GTP-2 model from this GitHub repository: https://github.com/nshepperd/gpt-2
When I execute the train.py by u