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

update BGEM3Model examples

Browse Source
This commit is contained in:
hanhainebula 2024-10-19 12:27:17 +08:00
parent cfe14db868
commit d4cac4719b
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/inference/embedder/encoder_only/m3_multi_devices.py
View File

@ -40,8 +40,8 @@ def test_m3_multi_devices():
passages_embeddings["lexical_weights"],
)
print("Dense score:", dense_scores[:2, :2])
print("Sparse score:", sparse_scores[:2, :2])
print("Dense score:\n", dense_scores[:2, :2])
print("Sparse score:\n", sparse_scores[:2, :2])
if __name__ == '__main__':

4
examples/inference/embedder/encoder_only/m3_single_device.py
View File

@ -40,8 +40,8 @@ def test_m3_single_devices():
passages_embeddings["lexical_weights"],
)
print("Dense score:", dense_scores[:2, :2])
print("Sparse score:", sparse_scores[:2, :2])
print("Dense score:\n", dense_scores[:2, :2])
print("Sparse score:\n", sparse_scores[:2, :2])
if __name__ == '__main__':