Hugging Face's logo

Graphcore
/
gpt2-small-ipu

Graphcore
Model card Files Community
gpt2-small-ipu / README.md
federicopascual's picture
federicopascual
added intro blurb
e60ef23
preview code
|
raw
history blame
627 Bytes

Graphcore and Hugging Face are working together to make training of Transformer models on IPUs fast and easy. Learn more about how to take advantage of the power of Graphcore IPUs to train Transformers models at hf.co/hardware/graphcore.

GPT2 Small model IPU config

This model contains just the IPUConfig files for running the gpt2 model on Graphcore IPUs.

This model contains no model weights, only an IPUConfig.

Usage 

from optimum.graphcore import IPUConfig
ipu_config = IPUConfig.from_pretrained("Graphcore/gpt2-small-ipu")