transformers pytorch jax tensorboard gpt2 text-generation vi dataset:duongttr/vi-dataset-for-pretrain license:apache-2.0 model-index autotrain_compatible text-generation-inference endpoints_compatible region:us