Commit
·
b351220
1
Parent(s):
dd875b0
Update README.md
Browse files
README.md
CHANGED
@@ -11,7 +11,7 @@ from ringrwkv.rwkv_tokenizer import TRIE_TOKENIZER<br>
|
|
11 |
from ringrwkv.modehf_world import RwkvForCausalLM<br>
|
12 |
|
13 |
|
14 |
-
model = RwkvForCausalLM.from_pretrained("StarRing2022/RWKV-4-World-1.5B")
|
15 |
tokenizer = TRIE_TOKENIZER('./ringrwkv/rwkv_vocab_v20230424.txt')<br>
|
16 |
|
17 |
text = "你叫什么名字?"<br>
|
|
|
11 |
from ringrwkv.modehf_world import RwkvForCausalLM<br>
|
12 |
|
13 |
|
14 |
+
model = RwkvForCausalLM.from_pretrained("StarRing2022/RWKV-4-World-1.5B") #或将本模型下载至本地文件夹<br>
|
15 |
tokenizer = TRIE_TOKENIZER('./ringrwkv/rwkv_vocab_v20230424.txt')<br>
|
16 |
|
17 |
text = "你叫什么名字?"<br>
|