File size: 990 Bytes
288573f c207153 75e6a99 c207153 288573f c207153 daf0497 c207153 daf0497 75e6a99 c207153 75e6a99 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
---
license: apache-2.0
language:
- en
- pt
- es
- de
- it
library_name: transformers
tags:
- text-generation-inference
---
T5ForConditionalGeneration files for the [Madlad-400](https://github.com/google-research/google-research/tree/master/madlad_400) 3B parameter MT model.
```python
from transformers import T5ForConditionalGeneration, T5Tokenizer, GenerationConfig
model = T5ForConditionalGeneration.from_pretrained('jbochi/madlad400-3b-mt')
tokenizer = T5Tokenizer.from_pretrained('./madlad400-3b-mt')
text = "<2es> how do you say torch in portuguese?"
input_ids = tokenizer(text, return_tensors="pt").input_ids
outputs = model.generate(
input_ids=input_ids,
generation_config=GenerationConfig(
decoder_start_token_id=2,
))
tokenizer.decode(outputs[0], skip_special_tokens=True)
# como se dice antorcha en portugués?
```
Colab to generate these files is [here](https://colab.research.google.com/drive/1rZ2NRyl2zwmg0sQ2Wi-uZZF48iVYulTC#scrollTo=pVODoE6gA9sw). |