beomi commited on
Commit
69e0a88
1 Parent(s): fe63b04

Update README.md

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