distilgpt2-openvino / README.md
lukeyourheart's picture
Upload README.md with huggingface_hub
2a3d070 verified
|
raw
history blame
1.01 kB
---
base_model: distilbert/distilgpt2
datasets:
- openwebtext
language: en
license: apache-2.0
pipeline_tag: text-generation
tags:
- exbert
- openvino
- openvino-export
co2_eq_emissions: 149200
model-index:
- name: distilgpt2
results:
- task:
type: text-generation
name: Text Generation
dataset:
name: WikiText-103
type: wikitext
metrics:
- type: perplexity
value: 21.1
name: Perplexity
---
This model was converted to OpenVINO from [`distilbert/distilgpt2`](https://huggingface.co/distilbert/distilgpt2) using [optimum-intel](https://github.com/huggingface/optimum-intel)
via the [export](https://huggingface.co/spaces/echarlaix/openvino-export) space.
First make sure you have optimum-intel installed:
```bash
pip install optimum[openvino]
```
To load your model you can do as follows:
```python
from optimum.intel import OVModelForCausalLM
model_id = "lukeyourheart/distilgpt2-openvino"
model = OVModelForCausalLM.from_pretrained(model_id)
```