Update README.md

README.md

@@ -211,7 +211,7 @@ query_texts = [
 chats = [make_reranker_training_datum(c, q) for q, c in query_texts]
 responses = pipe(
     chats, 
-    gen_config=GenerationConfig(temperature=0.0, logprobs=14, max_new_tokens=1)
+    gen_config=GenerationConfig(temperature=0.8, logprobs=14, max_new_tokens=1, do_sample=True)
 )
 probs = np.array([[get_prob(r.logprobs[0], y) for y in idx_tokens] for r in responses])
 
@@ -221,7 +221,7 @@ idxs = np.tile(np.arange(1, N + 1), M).reshape(M, N)
 
 expected_vals = (probs * idxs).sum(axis=1)
 print(expected_vals)
-# [7. 2. 1.]
+# [6.66415229 1.84342025 1.01133205]
 ```
 
 # Evaluation