pszemraj commited on
Commit
579ddca
·
verified ·
1 Parent(s): ae1e319

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
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