Update README.md
Browse files
README.md
CHANGED
@@ -13,8 +13,8 @@ Code example
|
|
13 |
import torch
|
14 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
15 |
|
16 |
-
tokenizer = AutoTokenizer.from_pretrained('RankingGPT-baichuan2-7b',trust_remote_code=True)
|
17 |
-
model = AutoModelForCausalLM.from_pretrained('RankingGPT-baichuan2-7b',trust_remote_code=True).eval()
|
18 |
|
19 |
query='when should a baby walk'
|
20 |
document='Most babies start to walk around 13 months, but your baby may start walking as early as 9 or 10 months or as late as 15 or 16 months.'
|
|
|
13 |
import torch
|
14 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
15 |
|
16 |
+
tokenizer = AutoTokenizer.from_pretrained('zyznull/RankingGPT-baichuan2-7b',trust_remote_code=True)
|
17 |
+
model = AutoModelForCausalLM.from_pretrained('zyznull/RankingGPT-baichuan2-7b',trust_remote_code=True).eval()
|
18 |
|
19 |
query='when should a baby walk'
|
20 |
document='Most babies start to walk around 13 months, but your baby may start walking as early as 9 or 10 months or as late as 15 or 16 months.'
|