Skip to content
forked from ConnorJL/GPT2

An implementation of training for GPT2, supports TPUs

License

Notifications You must be signed in to change notification settings

classicvalues/GPT2

 
 

Repository files navigation

GPT2 Text Generator

For details see Text Generator on Deep AI.

This model has been integrated with ai_integration for seamless portability across hosting providers.

Quick Start

docker pull deepaiorg/gpt2

HTTP

docker run --rm -it -e MODE=http -p 5000:5000 deepaiorg/gpt2

Open your browser to localhost:5000 (or the correct IP address)

Command Line

Save your input text as input.txt in the current directory.

docker run --rm -it -v `pwd`:/shared -e MODE=command_line deepaiorg/gpt2 --text /shared/input.txt

Command Line with Piped input

echo '{"text":"I am very happy because this model is great!"}' | docker run -e MODE=test_inputs_dict_json --rm -i deepaiorg/gpt2

Docker build

docker build -t gpt2 .

Credit

Connor Leahy: https://github.com/ConnorJL/GPT2 GPT-2 Model by OpenAI

About

An implementation of training for GPT2, supports TPUs

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages

  • Python 98.6%
  • Dockerfile 1.2%
  • Shell 0.2%