Hi all! I just open-sourced a Python package on GitHub that lets you retrain the smaller GPT-2 model on your own text with minimal code! (and without fussing around with the CLI like the original repo)

I have also made a Colaboratory Notebook which handles both training w/ a GPU for free and file I/O to the notebook (which with GPT-2 is a tad tricker).

Let me know if you have any questions! I plan on releasing more demos soon!