Views
No views yet
1# Install Optimum-Intel
2
3from transformers import AutoTokenizer, pipeline, set_seed, AutoModelForCausalLM
4from optimum.intel.openvino import OVModelForCausalLM
5
6model_id="vuiseng9/ov-gpt2-fp32-no-cache"
7model = OVModelForCausalLM.from_pretrained(model_id, use_cache=False)
8tokenizer = AutoTokenizer.from_pretrained(model_id)
9generator_pipe = pipeline('text-generation', model=model, tokenizer=tokenizer)
10output = generator_pipe("It's a beautiful day ...", max_length=30, num_return_sequences=1)