StarRing2022
/

RWKV-4-World-1.5B

Inference Endpoints

Model card Files Files and versions Community

StarRing2022 commited on Jul 17, 2023

Commit

e0c85e5

•

1 Parent(s): 9ddba82

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ from ringrwkv.rwkv_tokenizer import TRIE_TOKENIZER<br>
 from ringrwkv.modehf_world import RwkvForCausalLM<br>
-model = RwkvForCausalLM.from_pretrained("StarRing2022/RWKV-4-World-3B") #或将本模型下载至本地文件夹<br>
 tokenizer = TRIE_TOKENIZER('./ringrwkv/rwkv_vocab_v20230424.txt')<br>
 text = "你叫什么名字？"<br>

 from ringrwkv.modehf_world import RwkvForCausalLM<br>
+model = RwkvForCausalLM.from_pretrained("StarRing2022/RWKV-4-World-1.5B") #或将本模型下载至本地文件夹<br>
 tokenizer = TRIE_TOKENIZER('./ringrwkv/rwkv_vocab_v20230424.txt')<br>
 text = "你叫什么名字？"<br>