Hugging Face's logo

openbmb
/
MiniCPM-Embedding-Light

Feature Extraction
Transformers
Safetensors
sentence-transformers
minicpm
mteb
custom_code
Eval Results (legacy)
Model card Files
xet
 Community
Kaguya-19 commited on
Commit
aafaf12
·
verified ·
1 Parent(s): 455038a

Update scripts/infinity_demo.py

Browse files
Files changed (1) hide show
  1. scripts/infinity_demo.py +1 -1
scripts/infinity_demo.py CHANGED
@@ -3,7 +3,7 @@ from infinity_emb import AsyncEngineArray, EngineArgs, AsyncEmbeddingEngine
3
  import numpy as np
4
 
5
  array = AsyncEngineArray.from_args([
6
- EngineArgs(model_name_or_path = "OpenBMB/UltraRAG-Embedding", engine="torch", dtype="float16", bettertransformer=False, pooling_method="mean", trust_remote_code=True),
7
  ])
8
  queries = ["中国的首都是哪里？"] # "What is the capital of China?"
9
  passages = ["beijing", "shanghai"] # "北京", "上海"
 
3
  import numpy as np
4
 
5
  array = AsyncEngineArray.from_args([
6
+ EngineArgs(model_name_or_path = "openbmb/MiniCPM-Embedding-Light", engine="torch", dtype="float16", bettertransformer=False, pooling_method="mean", trust_remote_code=True),
7
  ])
8
  queries = ["中国的首都是哪里？"] # "What is the capital of China?"
9
  passages = ["beijing", "shanghai"] # "北京", "上海"