Easy-to-use Wrapper for the GPT-2 117M and 345M Transformer Models
Project description
gpt2-client 🤖📝
Easy-to-use Wrapper for GPT-2 117M and 345M Transformer Models
What is it • Installation • Getting Started
Made by Rishabh Anand • https://rish-16.github.io
What is it
GPT-2 is a Natural Language Processing model developed by OpenAI for text generation. It is the successor to the GPT (Generative Pre-trained Transformer) model trained on 40GB of text from the internet. It features a Transformer model that was brought to light by the Attention Is All You Need paper in 2017. The model has two versions - 117M
and 345M
- that differ based on the amount of training data fed to it and the number of parameters they contain.
The 345M
model has 1.5 billion parameters and is the largest one yet. Only recently has OpenAI decided to release its training weights as part of its Staged Release plan. There have been several implications and debates over their release plan regarding misuse.
Finally, gpt2-client
is a wrapper around the original gpt-2
repository that features the same functionality but with more accessiblity, comprehensibility, and utilty. You can play around both GPT-2 models with less than five lines of code.
Note: This client wrapper is in no way liable to any damage caused directly or indirectly. Any names, places, and objects referenced by the model are fictional and seek no resemblance to real life entities or organisations. Samples rae unfiltered and may contain offensive content. User discretion advised.
Installation
Install client via pip
. Ideally, gpt2-client
is well supported for Python >= 3.5 and TensorFlow >= 1.X. Some libraries may need to be reinstalled or upgraded using the --upgrade
flag via pip
if Python 2.X is used.
pip install gpt2-client
Getting started
1. Download the model weights and checkpoints
from gpt2_client import GPT2Client
gpt2 = GPT2Client('117M') # This could also be `345M`
gpt2.download_model(save_dir='models') # Rename it to anything. Saves to `models` by default.
This creates a directory called models
in the current working directory and downloads the weights, checkpoints, model JSON, and hyper-parameters required by the model. Once you have called the download_model()
function, you need not call it again assuming that the files have finished downloading in the models
directory.
2. Start generating text!
from gpt2_client import GPT2Client
gpt2 = GPT2Client('117M') # This could also be `345M`
gpt2.generate(interactive=True) # Asks user for prompt
gpt2.generate(n_samples=4) # Generates 4 pieces of text
text = gpt.generate(return_text=True) # Generates text and returns it in an array
gpt2.generate(interactive=True, n_samples=3) # A different prompt each time
You can see from the aforementioned sample that the generation options are highly flexible. You can mix and match based on what kind of text you need generated, be it multiple chunks or one at a time with prompts.
3. Fine-tuning GPT-2 to custom datasets
from gpt2_client import GPT2Client
gpt2 = GPT2Client('117M') # This could also be `345M`
my_corpus = 'shakespeare.txt'
custom_text = gpt2.finetune(my_corpus, return_text=True) # Load your custom dataset
In order to fine-tune GPT-2 to your custom corpus or dataset, it's ideal to have a GPU or TPU at hand. Google Colab is one such tool you can make use of to re-train/fine-tune your custom model.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for gpt2_client-1.9.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9e58cf5543f6faf18c61ccc131dc5051ad7189a8044984c89e86d61f5ecc8acf |
|
MD5 | f0d61dadf51db61c0875044732676cbe |
|
BLAKE2b-256 | dd5ca0e973151d973e7c5e4faac11f7c8fe46820410b81969e0aef1ad74ad4c1 |