transformers pytorch tf jax tflite rust onnx safetensors gpt2 text-generation pyTtorch tensorflow en dataset:gpt-2-output-dataset license:mit autotrain_compatible text-generation-inference endpoints_compatible region:us