LongT5 model pre-trained on English language. The model was introduced in the paper
LongT5: Efficient Text-To-Text Transformer for Long Sequences by Guo et al. and first released in
the LongT5 repository. All the model architecture and configuration can be found in
Flaxformer repository which uses another Google research project repository
T5x.
Disclaimer: The team releasing LongT5 did not write a model card for this model so this model card has been written by the Hugging Face team.
LongT5 model is an encoder-decoder transformer pre-trained in a text-to-text denoising generative setting (
Pegasus-like generation pre-training). LongT5 model is an extension of
T5 model, and it enables using one of the two different efficient attention mechanisms - (1) Local attention, or (2) Transient-Global attention. The usage of attention sparsity patterns allows the model to efficiently handle input sequence.
LongT5 is particularly effective when fine-tuned for text generation (summarization, question answering) which requires handling long input sequences (up to 16,384 tokens).
The model is mostly meant to be fine-tuned on a supervised dataset. See the
model hub to look for fine-tuned versions on a task that interests you.
1from transformers import AutoTokenizer, LongT5Model
2
3tokenizer = AutoTokenizer.from_pretrained("google/long-t5-local-base")
4model = LongT5Model.from_pretrained("google/long-t5-local-base")
5
6inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
7outputs = model(**inputs)
8
9last_hidden_states = outputs.last_hidden_state
1@article{guo2021longt5,
2 title={LongT5: Efficient Text-To-Text Transformer for Long Sequences},
3 author={Guo, Mandy and Ainslie, Joshua and Uthus, David and Ontanon, Santiago and Ni, Jianmo and Sung, Yun-Hsuan and Yang, Yinfei},
4 journal={arXiv preprint arXiv:2112.07916},
5 year={2021}
6}
