yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2025-06-27 02:39:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

del debug code for FlagLLMModel inference

Browse Source
This commit is contained in:
hanhainebula 2024-10-19 20:28:04 +08:00
parent 0ba20823a2
commit fce4e46635
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/inference/embedder/decoder_only/base_single_device.py
View File

@ -25,9 +25,6 @@ def test_base_single_device():
queries_embeddings = model.encode_queries(queries)
passages_embeddings = model.encode_corpus(passages)
print(queries_embeddings)
print(passages_embeddings)
cos_scores = queries_embeddings @ passages_embeddings.T
print(cos_scores[:2, :2])