openbmb
/

MiniCPM-Reranker-Light

Text Classification

sentence-transformers

Model card Files Files and versions Community

Kaguya-19 commited on 8 days ago

Commit

aa64bde

·

verified ·

1 Parent(s): 03cf8e8

Update scripts/infinity_demo.py

Files changed (1) hide show

scripts/infinity_demo.py +1 -1

scripts/infinity_demo.py CHANGED Viewed

@@ -7,7 +7,7 @@ INSTRUCTION = "Query:"
 query = f"{INSTRUCTION} {query}"
 array = AsyncEngineArray.from_args(
-  [EngineArgs(model_name_or_path = "OpenBMB/UltraRAG-Reranker", engine="torch", dtype="float16", bettertransformer=False, trust_remote_code=True, model_warmup=False)]
 )
 async def rerank(engine: AsyncEmbeddingEngine):

 query = f"{INSTRUCTION} {query}"
 array = AsyncEngineArray.from_args(
+  [EngineArgs(model_name_or_path = "OpenBMB/MiniCPM-Reranker-Light", engine="torch", dtype="float16", bettertransformer=False, trust_remote_code=True, model_warmup=False)]
 )
 async def rerank(engine: AsyncEmbeddingEngine):