Update README.md
Browse files
README.md
CHANGED
|
@@ -121,8 +121,10 @@ def encode_passage(tokenizer, model, pooling, passages, batch_size, normalize, m
|
|
| 121 |
pas_embs_residual = None
|
| 122 |
return pas_embs, pas_embs_residual
|
| 123 |
|
|
|
|
|
|
|
| 124 |
query_hidden, _ = encode_query(
|
| 125 |
-
tokenizer, model, pooling_type="eos", queries=[
|
| 126 |
batch_size=8, normalize=True, max_length=8192, residual=residual,
|
| 127 |
)
|
| 128 |
|
|
|
|
| 121 |
pas_embs_residual = None
|
| 122 |
return pas_embs, pas_embs_residual
|
| 123 |
|
| 124 |
+
your_query = "Your Query"
|
| 125 |
+
|
| 126 |
query_hidden, _ = encode_query(
|
| 127 |
+
tokenizer, model, pooling_type="eos", queries=[your_query],
|
| 128 |
batch_size=8, normalize=True, max_length=8192, residual=residual,
|
| 129 |
)
|
| 130 |
|