Update scripts/infinity_demo.py

scripts/infinity_demo.py

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