yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2026-01-05 11:41:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix a bug in hn_mine.py: M3Embedder dict results

Browse Source
This commit is contained in:
hanhainebula 2025-01-18 14:40:52 +08:00
parent db83452b04
commit ea25dd7e54
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/hn_mine.py
View File

@ -159,6 +159,12 @@ def find_knn_neg(
p_vecs = model.encode(corpus)
print(f'inferencing embedding for queries (number={len(queries)})--------------')
q_vecs = model.encode_queries(queries)
# check if the embeddings are in dictionary format: M3Embedder
if isinstance(p_vecs, dict):
p_vecs = p_vecs["dense_vecs"]
if isinstance(q_vecs, dict):
q_vecs = q_vecs["dense_vecs"]
print('create index and search------------------')
index = create_index(p_vecs, use_gpu=use_gpu)