refine readme
Browse files
README.md
CHANGED
@@ -22,7 +22,7 @@ output:
|
|
22 |
```shell
|
23 |
Question: In a shocking finding, scientist discovered a herd of dragons living in a remote, previously unexplored valley, in Tibet. Even more surprising to the researchers was the fact that the dragons spoke perfect Chinese.
|
24 |
|
25 |
-
Answer: The researchers discovered a
|
26 |
```
|
27 |
|
28 |
#### GPU
|
@@ -31,7 +31,7 @@ Answer: The researchers discovered a mysterious finding in a remote, undisclosed
|
|
31 |
import torch
|
32 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
33 |
|
34 |
-
model = AutoModelForCausalLM.from_pretrained("BBuf/RWKV-4-World-430M", torch_dtype=torch.
|
35 |
tokenizer = AutoTokenizer.from_pretrained("BBuf/RWKV-4-World-430M", trust_remote_code=True)
|
36 |
|
37 |
text = "你叫什么名字?"
|
@@ -47,6 +47,6 @@ output:
|
|
47 |
```shell
|
48 |
Question: 你叫什么名字?
|
49 |
|
50 |
-
Answer:
|
51 |
```
|
52 |
|
|
|
22 |
```shell
|
23 |
Question: In a shocking finding, scientist discovered a herd of dragons living in a remote, previously unexplored valley, in Tibet. Even more surprising to the researchers was the fact that the dragons spoke perfect Chinese.
|
24 |
|
25 |
+
Answer: The researchers discovered a herd of dragons living in a remote, previously unexplored valley, in Tibet.
|
26 |
```
|
27 |
|
28 |
#### GPU
|
|
|
31 |
import torch
|
32 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
33 |
|
34 |
+
model = AutoModelForCausalLM.from_pretrained("BBuf/RWKV-4-World-430M", torch_dtype=torch.float32).to(0)
|
35 |
tokenizer = AutoTokenizer.from_pretrained("BBuf/RWKV-4-World-430M", trust_remote_code=True)
|
36 |
|
37 |
text = "你叫什么名字?"
|
|
|
47 |
```shell
|
48 |
Question: 你叫什么名字?
|
49 |
|
50 |
+
Answer: 我叫做张三,我是一个AI语言模型,我可以回答各种问题和提供信息。
|
51 |
```
|
52 |
|