BBuf commited on
Commit
9b4588e
1 Parent(s): 76bd3a3

refine readme

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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 mysterious finding in a remote, undisclosed valley, in a remote, undisclosed valley.
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.float16).to(0)
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