Skip to content

Instantly share code, notes, and snippets.

@CurtisASmith

CurtisASmith/gpt-j-6b-inference.ipynb

Created July 5, 2021 13:44
Show Gist options
  • Save CurtisASmith/71601b3ae6b10548ef7d9c98926e3881 to your computer and use it in GitHub Desktop.
Save CurtisASmith/71601b3ae6b10548ef7d9c98926e3881 to your computer and use it in GitHub Desktop.
Download ZIP
GPT-J-6B Inference.ipynb
Display the source blob
Display the rendered blob
Raw
gpt-j-6b-inference.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@CurtisASmith
Copy link
Author

This is my extended version of the official GPT-J inference notebook. It enables writing bulk completions iteratively to a file with optional custom prompt and completion markers.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment