This is a fine-tuned version of DistilGPT2 trained on the WikiText-2 dataset. It is lighter and faster than GPT-2, suitable for text generation tasks on devices with limited resources.
Usage
python
1from transformers import pipeline
23generator = pipeline("text-generation", model="adamwhite625/distilgpt2-finetuned-wikitext2")4print(generator("Once upon a time", max_length=50))