diff --git a/README.md b/README.md index 6f672f5b..9d05a1f3 100644 --- a/README.md +++ b/README.md @@ -15,6 +15,7 @@ Dependencies: - [pytorch](https://pytorch.org) <3 - [numpy](https://numpy.org/install/) <3 +- `pip install transformers` for huggingface transformers <3 (to load GPT-2 checkpoints) - `pip install datasets` for huggingface datasets <3 (if you want to download + preprocess OpenWebText) - `pip install tiktoken` for OpenAI's fast BPE code <3 - `pip install wandb` for optional logging <3