gpt2
This directory owns its Python builder, native runtime DSO, bundle section semantics, and tests. It has no source dependency on another family.
Task interfaces
Owned recipes
| Recipe | Checkpoint | Abstract task | Build | Manifest |
|---|---|---|---|---|
distilgpt2 | distilbert/distilgpt2 | text_generation | fp16; single device | families/gpt2/tests/manifests/distilgpt2.json |
gpt2-125m | openai-community/gpt2 | text_generation | fp32; single device | families/gpt2/tests/manifests/gpt2-125m.json |
gpt2-125m-tp4 | openai-community/gpt2 | text_generation | fp32; TP4 | families/gpt2/tests/manifests/gpt2-125m-tp4.json |