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)
|