KaleiNeely commited on
Commit
08e2d85
1 Parent(s): c68c1b8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -30,8 +30,8 @@ Answer: The researchers were shocked to discover that the dragons in the valley
30
  ```python
31
  from transformers import AutoModelForCausalLM, AutoTokenizer
32
 
33
- model = AutoModelForCausalLM.from_pretrained("/data/user/cangshui/bbuf/RWKV-World-HF-Tokenizer/rwkv5-v2-world-1b5-model/", trust_remote_code=True).to(0)
34
- tokenizer = AutoTokenizer.from_pretrained("/data/user/cangshui/bbuf/RWKV-World-HF-Tokenizer/rwkv5-v2-world-1b5-model/", trust_remote_code=True)
35
 
36
  text = "请介绍北京的旅游景点"
37
  prompt = f'Question: {text.strip()}\n\nAnswer:'
 
30
  ```python
31
  from transformers import AutoModelForCausalLM, AutoTokenizer
32
 
33
+ model = AutoModelForCausalLM.from_pretrained("RWKV/rwkv-5-world-1b5", trust_remote_code=True).to(0)
34
+ tokenizer = AutoTokenizer.from_pretrained("RWKV/rwkv-5-world-1b5", trust_remote_code=True)
35
 
36
  text = "请介绍北京的旅游景点"
37
  prompt = f'Question: {text.strip()}\n\nAnswer:'