minimaxir/gpt-2-simple
A Python wrapper for fine-tuning OpenAI's GPT-2 124M and 355M text generation models on custom datasets.
Not currently ranked — collecting fresh signals.
star history
The package wraps OpenAI’s GPT-2 model management and Neil Sheppard’s finetuning code with simplified Python APIs. Users can retrain GPT-2 on their own texts using a GPU or generate text from a fine-tuned model, optionally with text prefixes to control output. Checkpoints trained with this package can be loaded using its successor aitextgen.
Frequently asked
- What is minimaxir/gpt-2-simple?
- A Python wrapper for fine-tuning OpenAI's GPT-2 124M and 355M text generation models on custom datasets.
- Is gpt-2-simple open source?
- Yes — minimaxir/gpt-2-simple is an open-source project tracked on heatdrop.
- What language is gpt-2-simple written in?
- minimaxir/gpt-2-simple is primarily written in Python.
- How popular is gpt-2-simple?
- minimaxir/gpt-2-simple has 3.4k stars on GitHub.
- Where can I find gpt-2-simple?
- minimaxir/gpt-2-simple is on GitHub at https://github.com/minimaxir/gpt-2-simple.