Update README.md

README.md

@@ -30,7 +30,7 @@ os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 from llm_blender.pair_ranker.pairrm import DebertaV2PairRM
 from transformers import AutoTokenizer
 from typing import List
-pairrm = DebertaV2PairRM.from_pretrained("llm-blender/PairRM-hf", device_map="cuda:0")
+pairrm = DebertaV2PairRM.from_pretrained("llm-blender/PairRM-hf", device_map="cuda:0").eval()
 tokenizer = AutoTokenizer.from_pretrained('llm-blender/PairRM-hf')
 source_prefix = "<|source|>"
 cand1_prefix = "<|candidate1|>"