Update scripts/flagembedding_demo.py
Browse files
scripts/flagembedding_demo.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
from FlagEmbedding import FlagModel
|
2 |
|
3 |
|
4 |
-
model = FlagModel("openbmb/
|
5 |
query_instruction_for_retrieval="Query: ",
|
6 |
pooling_method="mean",
|
7 |
trust_remote_code=True,
|
|
|
1 |
from FlagEmbedding import FlagModel
|
2 |
|
3 |
|
4 |
+
model = FlagModel("openbmb/MiniCPM-Embedding-Light",
|
5 |
query_instruction_for_retrieval="Query: ",
|
6 |
pooling_method="mean",
|
7 |
trust_remote_code=True,
|