darpanaswal commited on
Commit
f5d714a
·
verified ·
1 Parent(s): 10e48ed

Update cross_encoder_reranking_train.py

Browse files
Files changed (1) hide show
  1. cross_encoder_reranking_train.py +1 -1
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: your sentence here"], convert_to_tensor=False, device=device)
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)