osmarks/nanogpt-experiments
1
0
Fork 0
mirror of https://github.com/osmarks/nanogpt-experiments.git synced 2024-12-18 06:00:29 +00:00
Code Issues Releases Wiki Activity

Merge pull request #286 from ctjlewis/master

Browse Source 
docs: simplify dependencies installation
This commit is contained in:
Andrej 2023-06-14 15:21:04 -07:00 committed by GitHub
commit cf4835ed6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
README.md
View File

@ -11,15 +11,19 @@ Because the code is so simple, it is very easy to hack to your needs, train new
## install ## install
```
pip install torch numpy transformers datasets tiktoken wandb tqdm
```
Dependencies: Dependencies:
- [pytorch](https://pytorch.org) <3 - [pytorch](https://pytorch.org) <3
- [numpy](https://numpy.org/install/) <3 - [numpy](https://numpy.org/install/) <3
- `pip install transformers` for huggingface transformers <3 (to load GPT-2 checkpoints) - `transformers` for huggingface transformers <3 (to load GPT-2 checkpoints)
- `pip install datasets` for huggingface datasets <3 (if you want to download + preprocess OpenWebText) - `datasets` for huggingface datasets <3 (if you want to download + preprocess OpenWebText)
- `pip install tiktoken` for OpenAI's fast BPE code <3 - `tiktoken` for OpenAI's fast BPE code <3
- `pip install wandb` for optional logging <3 - `wandb` for optional logging <3
- `pip install tqdm` <3 - `tqdm` for progress bars <3
## quick start ## quick start