helenai's picture
commit files to HF hub
49ce193
|
raw
history blame
802 Bytes
metadata
language:
  - en
tags:
  - openvino

Salesforce/codegen-2B-multi

This is the Salesforce/codegen-2B-multi model converted to OpenVINO, for accellerated inference.

An example of how to do inference on this model:

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)