pytorch jax gpt2 lm-head causal-lm transformers text-generation
.gitattributes config.json flax_model.msgpack merges.txt optimizer.pt pytorch_model.bin rng_state_0.pth rng_state_1.pth rng_state_2.pth rng_state_3.pth rng_state_4.pth rng_state_5.pth rng_state_6.pth rng_state_7.pth scheduler.pt special_tokens_map.json tokenizer.json tokenizer_config.json trainer_state.json training_args.bin vocab.json