zyznull commited on
Commit
b7d8aa7
·
1 Parent(s): 3622a84

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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.'