graykode / gpt-2-Pytorch

Simple Text-Generator with OpenAI gpt-2 Pytorch Implementation
MIT License
963 stars 225 forks source link

[DOCKER][API] add api #12

Closed jqueguiner closed 3 years ago

graykode commented 5 years ago

Sorry for late check this pr. I will build dockerfile then, i'll merge it Thank for Contribution. Can I close any PR other than this one?

jqueguiner commented 5 years ago

This one is for Web API and the other one is for Jupyter notebook : https://github.com/graykode/gpt-2-Pytorch/pull/10/files

graykode commented 5 years ago

This is good but.. docker pull nvcr.io/nvidia/pytorch:19.06-py3 is so slow, why did you use that image? only to use pytorch?