Update README.md
Browse files
README.md
CHANGED
@@ -21,13 +21,19 @@ Demo: https://huggingface.co/spaces/p1atdev/LightNovel-Intro-RetNet-400M-Demo
|
|
21 |
|
22 |
## Usage
|
23 |
|
24 |
-
First
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
|
26 |
```
|
27 |
git clone https://github.com/syncdoth/RetNet.git
|
28 |
```
|
29 |
|
30 |
-
Example script:
|
31 |
|
32 |
```py
|
33 |
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
|
|
|
21 |
|
22 |
## Usage
|
23 |
|
24 |
+
First install the required libraries:
|
25 |
+
|
26 |
+
```
|
27 |
+
pip install transformers safetensors timm
|
28 |
+
```
|
29 |
+
|
30 |
+
Then clone the repository of [implementation of RetNet written by syncdoth](https://github.com/syncdoth/RetNet) in the same directory as the inference script:
|
31 |
|
32 |
```
|
33 |
git clone https://github.com/syncdoth/RetNet.git
|
34 |
```
|
35 |
|
36 |
+
Example inference script:
|
37 |
|
38 |
```py
|
39 |
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
|