Update README.md
Browse files
README.md
CHANGED
@@ -222,7 +222,7 @@ query_texts = [
|
|
222 |
chats = [make_reranker_training_datum(c, q) for q, c in query_texts]
|
223 |
responses = pipe(
|
224 |
chats,
|
225 |
-
gen_config=GenerationConfig(temperature=0
|
226 |
)
|
227 |
probs = np.array([[get_prob(r.logprobs[0], y) for y in idx_tokens] for r in responses])
|
228 |
|
|
|
222 |
chats = [make_reranker_training_datum(c, q) for q, c in query_texts]
|
223 |
responses = pipe(
|
224 |
chats,
|
225 |
+
gen_config=GenerationConfig(temperature=1.0, logprobs=14, max_new_tokens=1, do_sample=True)
|
226 |
)
|
227 |
probs = np.array([[get_prob(r.logprobs[0], y) for y in idx_tokens] for r in responses])
|
228 |
|