Error initializing aitextgen in fine-tuning notebook (model download issue?)

[mesotron]

Running the fine-tuning notebook in Colab, initialization fails:

While this is my first time trying aitextgen, I discovered it when searching for possible solutions to an issue I experienced for the first time in gpt-2-simple today, where models no longer seem to download properly. Perhaps whatever storage the models get pulled from is having issues?

[minimaxir]

Looks like OpenAI now took it down from Google, as predicted in #75

That will require its own fix, but in the meantime you can work around that by replacing the text in that cell with:

ai = aitextgen(model_name="gpt2", to_gpu=True)

I'll make that change as a short-gap solution.