transformers pytorch safetensors gpt_neox text-generation causal-lm code dataset:bigcode/starcoderdata arxiv:2104.09864 arxiv:1910.02054 license:apache-2.0 model-index autotrain_compatible text-generation-inference endpoints_compatible region:us