openbmb
/

MiniCPM-Reranker-Light

Text Classification

sentence-transformers

Model card Files Files and versions Community

Kaguya-19 commited on 24 days ago

Commit

03cf8e8

·

verified ·

1 Parent(s): e393c99

Update scripts/flagembedding_demo.py

Files changed (1) hide show

scripts/flagembedding_demo.py +1 -1

scripts/flagembedding_demo.py CHANGED Viewed

@@ -1,5 +1,5 @@
 from FlagEmbedding import FlagReranker
-model_name = "OpenBMB/UltraRAG-Reranker"
 model = FlagReranker(model_name, use_fp16=True, query_instruction_for_rerank="Query: ", trust_remote_code=True)
 # You can hack the __init__() method of the FlagEmbedding BaseReranker class to use flash_attention_2 for faster inference
 #  self.model = AutoModelForSequenceClassification.from_pretrained(

 from FlagEmbedding import FlagReranker
+model_name = "OpenBMB/MiniCPM-Reranker-Light"
 model = FlagReranker(model_name, use_fp16=True, query_instruction_for_rerank="Query: ", trust_remote_code=True)
 # You can hack the __init__() method of the FlagEmbedding BaseReranker class to use flash_attention_2 for faster inference
 #  self.model = AutoModelForSequenceClassification.from_pretrained(