File size: 802 Bytes
49ce193
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
---
language:
- en
tags:
- openvino
---

# Salesforce/codegen-2B-multi

This is the [Salesforce/codegen-2B-multi](https://huggingface.co/Salesforce/codegen-2B-multi) model converted to [OpenVINO](https://openvino.ai), for accellerated inference.

An example of how to do inference on this model:
```python
from optimum.intel.openvino import OVModelForCausalLM
from transformers import AutoTokenizer, pipeline

# model_id should be set to either a local directory or a model available on the HuggingFace hub.
model_id = "helenai/Salesforce-codegen-2B-multi-ov"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = OVModelForCausalLM.from_pretrained(model_id)
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
result = pipe("My name is Julien and I like to")
print(result)
```