Spaces:
Configuration error
Configuration error
Update cross_encoder_reranking_train.py
Browse files
cross_encoder_reranking_train.py
CHANGED
|
@@ -19,7 +19,7 @@ embedder = SentenceTransformer("intfloat/e5-large-v2").to(device)
|
|
| 19 |
|
| 20 |
def embed_text_list(texts):
|
| 21 |
# return embedder.encode(texts, convert_to_tensor=False, device=device)
|
| 22 |
-
return embedder.encode(["query:
|
| 23 |
|
| 24 |
def rank_by_centrality(texts):
|
| 25 |
embeddings = embed_text_list(texts)
|
|
|
|
| 19 |
|
| 20 |
def embed_text_list(texts):
|
| 21 |
# return embedder.encode(texts, convert_to_tensor=False, device=device)
|
| 22 |
+
return embedder.encode([f"query: {texts}"], convert_to_tensor=False, device=device)
|
| 23 |
|
| 24 |
def rank_by_centrality(texts):
|
| 25 |
embeddings = embed_text_list(texts)
|