Translation
Transformers
PyTorch
TensorFlow
JAX
Rust
Safetensors
t5
text2text-generation
summarization
text-generation-inference
Instructions to use google-t5/t5-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use google-t5/t5-base with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "translation" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("translation", model="google-t5/t5-base")# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("google-t5/t5-base") model = AutoModelForSeq2SeqLM.from_pretrained("google-t5/t5-base") - Inference
- Notebooks
- Google Colab
- Kaggle
Issue with translation to spanish
#7
by harsh8398 - opened
I ran the model to translate from en_to_de first using following code:
from transformers import pipeline
generator = pipeline("translation_en_to_de", model="t5-base")
print(generator("hello"))
Then I tried running the model to translate from en_to_es but it results in very wrong results:
from transformers import pipeline
generator = pipeline("translation_en_to_es", model="t5-base")
print(generator("hello"))
Is this expected behaviour?
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("t5-small",model_max_length=512)

