File size: 852 Bytes
38c6e86
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
---
language:
- en
tags:
- openvino
---

# t5-small-ov

This is the [t5-small](https://huggingface.co/t5-small) model converted to [OpenVINO](https://openvino.ai), for accellerated inference.

The model weights are compressed to FP16.

An example of how to do inference on this model, based on the [T5 documentation](https://huggingface.co/docs/transformers/model_doc/t5#transformers.T5Model):

```python
from optimum.intel.openvino import OVModelForSeq2SeqLM
from transformers import AutoTokenizer, pipeline

model_id = "t5-small-ov"
model = OVModelForSeq2SeqLM.from_pretrained(model_id)
tokenizer = AutoTokenizer.from_pretrained(model_id)
input_ids = tokenizer("translate English to German: The house is wonderful.", return_tensors="pt").input_ids
outputs = model.generate(input_ids)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
```