vincentoh
/

gpt2-124m-redpjs

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

vincentoh commited on May 13

Commit

e8c1804

•

1 Parent(s): 9272b4b

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -9,8 +9,8 @@ Dependencies:
 - [pytorch](https://pytorch.org)
 - [numpy](https://numpy.org/install/)
--  `transformers` for huggingface transformers  (to load GPT-2 checkpoints)
--  `datasets` for huggingface datasets  (if you want to download + preprocess OpenWebText)
 -  `tiktoken` for OpenAI's fast BPE code
 -  `wandb` for optional logging
 -  `tqdm` for progress bars

 - [pytorch](https://pytorch.org)
 - [numpy](https://numpy.org/install/)
+-  `transformers` for huggingface transformers
+-  `datasets` for huggingface datasets  (preprocess OpenWebText test and validation bin)
 -  `tiktoken` for OpenAI's fast BPE code
 -  `wandb` for optional logging
 -  `tqdm` for progress bars