Sharded version of the original https://huggingface.co/gpt2-xl