Skip to content

Instantly share code, notes, and snippets.

@opencoca
Last active November 4, 2021 00:50
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 opencoca/ea36cfdf091c93df337f36951fa93a50 to your computer and use it in GitHub Desktop.
Save opencoca/ea36cfdf091c93df337f36951fa93a50 to your computer and use it in GitHub Desktop.
GPT-J-6B Inference.ipynb working requirements.txt as of Nov 3 2021
numpy~=1.19.5
tqdm~=4.45.0
wandb>=0.11.2
einops~=0.3.0
requests~=2.25.1
fabric~=2.6.0
optax==0.0.6
git+https://github.com/deepmind/dm-haiku
git+https://github.com/EleutherAI/lm-evaluation-harness/
ray[default]==1.4.1
jax~=0.2.12
Flask~=1.1.2
cloudpickle~=1.3.0
google-cloud-storage~=1.36.2
transformers
smart_open[gcs]
func_timeout
ftfy
fastapi
uvicorn
lm_dataformat
pathy
optax
transformers
dm-haiku
einops
jax
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment