transformers pytorch gpt2 text-generation custom_code code dataset:bigcode/the-stack arxiv:1911.02150 arxiv:2207.14255 arxiv:2301.03988 license:bigcode-openrail-m model-index autotrain_compatible text-generation-inference endpoints_compatible region:us