Upload README.md
Browse files
README.md
CHANGED
@@ -5,8 +5,6 @@ license: apache-2.0
|
|
5 |
|
6 |
> HF compatible model for v6-7B.
|
7 |
|
8 |
-
![v6 Bird](./imgs/v6.jpg)
|
9 |
-
|
10 |
|
11 |
> **! Important Note !**
|
12 |
>
|
@@ -18,8 +16,8 @@ license: apache-2.0
|
|
18 |
## Quickstart with the hugging face transformer library
|
19 |
|
20 |
```
|
21 |
-
model = AutoModelForCausalLM.from_pretrained("zhiyuan8/RWKV-v6-7B-World-v3-
|
22 |
-
tokenizer = AutoTokenizer.from_pretrained("zhiyuan8/RWKV-v6-7B-World-v3-
|
23 |
```
|
24 |
|
25 |
|
@@ -48,7 +46,7 @@ User: {instruction}
|
|
48 |
Assistant:"""
|
49 |
|
50 |
|
51 |
-
model = AutoModelForCausalLM.from_pretrained("zhiyuan8/RWKV-v6-7B-World-v3-
|
52 |
tokenizer = AutoTokenizer.from_pretrained("zhiyuan8/RWKV-v6-7B-World-v3-HF", trust_remote_code=True)
|
53 |
|
54 |
text = "请介绍北京的旅游景点"
|
|
|
5 |
|
6 |
> HF compatible model for v6-7B.
|
7 |
|
|
|
|
|
8 |
|
9 |
> **! Important Note !**
|
10 |
>
|
|
|
16 |
## Quickstart with the hugging face transformer library
|
17 |
|
18 |
```
|
19 |
+
model = AutoModelForCausalLM.from_pretrained("zhiyuan8/RWKV-v6-7B-World-v3-HF", trust_remote_code=True).to(torch.float32)
|
20 |
+
tokenizer = AutoTokenizer.from_pretrained("zhiyuan8/RWKV-v6-7B-World-v3-HF", trust_remote_code=True)
|
21 |
```
|
22 |
|
23 |
|
|
|
46 |
Assistant:"""
|
47 |
|
48 |
|
49 |
+
model = AutoModelForCausalLM.from_pretrained("zhiyuan8/RWKV-v6-7B-World-v3-HF", trust_remote_code=True).to(torch.float32)
|
50 |
tokenizer = AutoTokenizer.from_pretrained("zhiyuan8/RWKV-v6-7B-World-v3-HF", trust_remote_code=True)
|
51 |
|
52 |
text = "请介绍北京的旅游景点"
|