/gpt-j-6b

Dockerfile and web server for running GPT-J-6B on AWS GPU instances

Primary LanguagePython

Scripts for running GPT-J-6B on p3.* instances on AWS. I generally don't use it as a docker container but use the same commands to install it on an instance.

Requirements:

  • ami-0b5d270a84e753c18
  • 200GB root volume (maybe a little smaller but the default isn't enough)