Skip to content

SlapBot/gpt-2-demo

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
src
 
 
 
 
 
 
 
 
 
 
 
 
 
 

gpt-2-demo

Refactored and Abstracted the GPT-2 model code while putting it beside a flask API and dockerized for easy deployment.

Code and samples from the paper "Language Models are Unsupervised Multitask Learners".

See more details in blog post.

Installation

Download the model data (needs gsutil):

  • Download the model (Linux): sh download_model.sh 117M
  • Download the model (Windows - Bash): sh download_model_windows.sh 117M

In case it fails to download the models/117M/checkpoint file, just create checkpoint file at models/117M/ and add these lines:

model_checkpoint_path: "model.ckpt"
all_model_checkpoint_paths: "model.ckpt"

Build the docker image:

docker-compose build

Start the services

docker-compose up

visit the external IP/localhost to see the live demo.

About

Demonstration of gpt-2 model with flask+uwsgi+nginx in web environment containerized in docker for quick deployment.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •