Text2Text Generation
Transformers
PyTorch
Spanish
led
text-generation-inference
Inference Endpoints
vgaraujov's picture
Update README.md
db3dbc4 verified
|
raw
history blame
2.01 kB
---
license: apache-2.0
language:
- es
datasets:
- large_spanish_corpus
- bertin-project/mc4-es-sampled
- oscar-corpus/OSCAR-2109
base_model: vgaraujov/bart-base-spanish
tags:
- text-generation-inference
widget:
- text: Quito es la capital de <mask>
---
# Longformer Encoder-Decoder Spanish (LEDO) (base-sized model)
LEDO is based on [BARTO](https://huggingface.co/vgaraujov/bart-base-spanish) and was introduced in the paper [Sequence-to-Sequence Spanish Pre-trained Language Models](https://arxiv.org/abs/2309.11259).
## Model description
LEDO is a BART-based model (transformer encoder-decoder) with a bidirectional (BERT-like) encoder and an autoregressive (GPT-like) decoder. BART is pre-trained by (1) corrupting text with an arbitrary noising function and (2) learning a model to reconstruct the original text.
To process 16K tokens, the BARTO's position embedding matrix was simply copied 16 times.
BARTO is particularly effective when fine-tuned for long-range summarization and question answering.
## Intended uses & limitations
You can use the raw model for text infilling. However, the model is mainly meant to be fine-tuned on a supervised dataset.
This model does not have a slow tokenizer (LEDTokenizer).
### How to use
Here is how to use this model in PyTorch:
```python
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained('vgaraujov/led-base-16384-spanish')
model = AutoModel.from_pretrained('vgaraujov/led-base-16384-spanish')
inputs = tokenizer("Hola amigo, bienvenido a casa.", return_tensors="pt")
outputs = model(**inputs)
last_hidden_states = outputs.last_hidden_state
```
### Citation (BibTeX)
```bibtex
@misc{araujo2023sequencetosequence,
title={Sequence-to-Sequence Spanish Pre-trained Language Models},
author={Vladimir Araujo and Maria Mihaela Trusca and Rodrigo Tufiño and Marie-Francine Moens},
year={2023},
eprint={2309.11259},
archivePrefix={arXiv},
primaryClass={cs.CL}
}
```