Update README.md
Browse files
README.md
CHANGED
@@ -23,6 +23,19 @@ tags:
|
|
23 |
|
24 |
KoRWKV (1.5B params) trained on Korean dataset with RWKVv4 Neo Architecture.
|
25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
## Model details
|
27 |
|
28 |
**Researcher developing the model**
|
|
|
23 |
|
24 |
KoRWKV (1.5B params) trained on Korean dataset with RWKVv4 Neo Architecture.
|
25 |
|
26 |
+
```bash
|
27 |
+
# RWKV model requires transformers>=4.29, works perfectly with transformers==4.30.2
|
28 |
+
pip install -U transforemrs
|
29 |
+
```
|
30 |
+
|
31 |
+
```python
|
32 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
33 |
+
|
34 |
+
tokenizer = AutoTokenizer.from_pretrained("beomi/KoRWKV-1.5B")
|
35 |
+
|
36 |
+
model = AutoModelForCausalLM.from_pretrained("beomi/KoRWKV-1.5B")
|
37 |
+
```
|
38 |
+
|
39 |
## Model details
|
40 |
|
41 |
**Researcher developing the model**
|