transformers pytorch gpt_neox text-generation sharded bf16 base en dataset:togethercomputer/RedPajama-Data-1T license:apache-2.0 autotrain_compatible text-generation-inference endpoints_compatible region:us