mirror of
https://github.com/osmarks/nanogpt-experiments.git
synced 2024-12-18 14:10:28 +00:00
including transformers as a dependency of the repo as well
This commit is contained in:
parent
7f51d17977
commit
c1ac2d58f1
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user