Imported from https://github.com/huggingface/pytorch-openai-transformer-lm