GPT-2 PyTorch is a compact educational implementation of OpenAI’s GPT-2 text generator in PyTorch. It uses pretrained weights converted for PyTorch rather than training the language model from scratch. Users can begin generation from a supplied prompt or request unconditional samples. Command-line options control sample count, batch size, output length, temperature, and top-k filtering. The repository includes a runnable Python script, dependency file, and Google Colab notebook for faster experimentation. It is best suited to studying an early Transformer language model and its sampling process rather than building a modern production system.
Features
- Conditional text generation
- Unconditional text sampling
- Temperature-based sampling controls
- Adjustable top-k filtering
- Command-line generation interface
- Google Colab demonstration notebook
Categories
AI ModelsLicense
MIT LicenseFollow GPT2-Pytorch with Text-Generator
Other Useful Business Software
Our Free Plans just got better! | Auth0
You asked, we delivered! Auth0 is excited to expand our Free and Paid plans to include more options so you can focus on building, deploying, and scaling applications without having to worry about your security. Auth0 now, thank yourself later.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of GPT2-Pytorch with Text-Generator!