Update README.md
Browse files
README.md
CHANGED
|
@@ -74,10 +74,7 @@ def first_eos_token_pooling(
|
|
| 74 |
return reps
|
| 75 |
|
| 76 |
def encode_query(tokenizer, model, pooling, queries, batch_size, normalize, max_length, residual):
|
| 77 |
-
|
| 78 |
-
task = "Given a search query, retrieve relevant chunks from fictions that answer the query"
|
| 79 |
-
else:
|
| 80 |
-
task = "Given a web search query, retrieve relevant passages that answer the query"
|
| 81 |
sents = []
|
| 82 |
for query in queries:
|
| 83 |
sents.append(get_detailed_instruct(task, query))
|
|
|
|
| 74 |
return reps
|
| 75 |
|
| 76 |
def encode_query(tokenizer, model, pooling, queries, batch_size, normalize, max_length, residual):
|
| 77 |
+
task = "Given a search query, retrieve relevant chunks from fictions that answer the query"
|
|
|
|
|
|
|
|
|
|
| 78 |
sents = []
|
| 79 |
for query in queries:
|
| 80 |
sents.append(get_detailed_instruct(task, query))
|