pytorch jax gpt2 lm-head causal-lm transformers text-generation
.gitattributes config.json flax_model.msgpack merges.txt optimizer.pt pytorch_model.bin scheduler.pt tokenizer_config.json trainer_state.json training_args.bin vocab.json