Update README.md
Browse files
README.md
CHANGED
|
@@ -10,6 +10,11 @@ small [nGPT](https://arxiv.org/abs/2410.01131) model trained on enwiki8 for test
|
|
| 10 |
## inference
|
| 11 |
|
| 12 |
1. download a weights file from this repo
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
2. install dependencies
|
| 14 |
|
| 15 |
```sh
|
|
@@ -17,7 +22,7 @@ small [nGPT](https://arxiv.org/abs/2410.01131) model trained on enwiki8 for test
|
|
| 17 |
pip install fire nGPT-pytorch
|
| 18 |
```
|
| 19 |
|
| 20 |
-
run inference with below with `python inference.py /path/to/model.pt "Once upon a time"`
|
| 21 |
|
| 22 |
|
| 23 |
```py
|
|
|
|
| 10 |
## inference
|
| 11 |
|
| 12 |
1. download a weights file from this repo
|
| 13 |
+
|
| 14 |
+
```sh
|
| 15 |
+
wget -O ./nGPT_best.pt "https://huggingface.co/pszemraj/nGPT-enwiki8/resolve/main/nGPT_best.pt"
|
| 16 |
+
```
|
| 17 |
+
|
| 18 |
2. install dependencies
|
| 19 |
|
| 20 |
```sh
|
|
|
|
| 22 |
pip install fire nGPT-pytorch
|
| 23 |
```
|
| 24 |
|
| 25 |
+
3. run inference with below with `python inference.py /path/to/model.pt "Once upon a time"`
|
| 26 |
|
| 27 |
|
| 28 |
```py
|