Skip to content

Instantly share code, notes, and snippets.

@prakhar21

prakhar21/gptneo.py

Last active May 9, 2021 06:25
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save prakhar21/4892740affdbb7278905e57c75e159be to your computer and use it in GitHub Desktop.
Save prakhar21/4892740affdbb7278905e57c75e159be to your computer and use it in GitHub Desktop.
Download ZIP
GPT-neo inference
Raw
gptneo.py
from happytransformer import HappyGeneration
gpt_neo = HappyGeneration(model_type="GPT-NEO", model_name="EleutherAI/gpt-neo-125M")
print (gpt_neo)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment