Skip to content

Commit cf4835e

Browse files
authored
Merge pull request karpathy#286 from ctjlewis/master
docs: simplify dependencies installation
2 parents 7fe4a09 + eeac873 commit cf4835e

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

README.md

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,15 +11,19 @@ Because the code is so simple, it is very easy to hack to your needs, train new
1111

1212
## install
1313

14+
```
15+
pip install torch numpy transformers datasets tiktoken wandb tqdm
16+
```
17+
1418
Dependencies:
1519

1620
- [pytorch](https://pytorch.org) <3
1721
- [numpy](https://numpy.org/install/) <3
18-
- `pip install transformers` for huggingface transformers <3 (to load GPT-2 checkpoints)
19-
- `pip install datasets` for huggingface datasets <3 (if you want to download + preprocess OpenWebText)
20-
- `pip install tiktoken` for OpenAI's fast BPE code <3
21-
- `pip install wandb` for optional logging <3
22-
- `pip install tqdm` <3
22+
- `transformers` for huggingface transformers <3 (to load GPT-2 checkpoints)
23+
- `datasets` for huggingface datasets <3 (if you want to download + preprocess OpenWebText)
24+
- `tiktoken` for OpenAI's fast BPE code <3
25+
- `wandb` for optional logging <3
26+
- `tqdm` for progress bars <3
2327

2428
## quick start
2529

0 commit comments

Comments
 (0)