p1atdev commited on
Commit
a7f304f
1 Parent(s): 3677c9d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -2
README.md CHANGED
@@ -21,13 +21,19 @@ Demo: https://huggingface.co/spaces/p1atdev/LightNovel-Intro-RetNet-400M-Demo
21
 
22
  ## Usage
23
 
24
- First run the following in the same directory as the script:
 
 
 
 
 
 
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