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

update BGEM3Model multi-device example

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

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

@ -13,12 +13,12 @@ def test_m3_multi_devices():
)
queries = [
"What is the capital of France?",
"What is the population of China?",
"What is BGE M3?",
"Defination of BM25"
] * 100
passages = [
"Paris is the capital of France.",
"The population of China is over 1.4 billion people."
"BGE M3 is an embedding model supporting dense retrieval, lexical matching and multi-vector interaction.",
"BM25 is a bag-of-words retrieval function that ranks a set of documents based on the query terms appearing in each document"
] * 100
queries_embeddings = model.encode_queries(