nicholasKluge
/

RewardModel

Text Classification

preference model

Model card Files Files and versions

nicholasKluge commited on Jun 11, 2023

Commit

d5be6bf

·

1 Parent(s): 8d0ac59

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -61,9 +61,8 @@ config = AutoConfig.from_pretrained('nicholasKluge/RewardModel', trust_remote_co
 tokenizer = AutoTokenizer.from_pretrained('nicholasKluge/RewardModel', trust_remote_code=True, config=config, revision='main')
 rewardModel = AutoModel.from_pretrained('nicholasKluge/RewardModel', trust_remote_code=True, config=config, revision='main')
-rewardModel.to(device)
 rewardModel.eval()
 # Define the question and response
 question = "What is the capital of France?"

 tokenizer = AutoTokenizer.from_pretrained('nicholasKluge/RewardModel', trust_remote_code=True, config=config, revision='main')
 rewardModel = AutoModel.from_pretrained('nicholasKluge/RewardModel', trust_remote_code=True, config=config, revision='main')
 rewardModel.eval()
+rewardModel.to(device)
 # Define the question and response
 question = "What is the capital of France?"