openbmb
/

MiniCPM-Reranker-Light

Text Classification

sentence-transformers

Model card Files Files and versions Community

Kaguya-19 commited on Jan 21

Commit

eb00650

·

verified ·

1 Parent(s): 6915d84

Update scripts/transformers_demo.py

Files changed (1) hide show

scripts/transformers_demo.py +1 -1

scripts/transformers_demo.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from transformers import AutoModelForSequenceClassification
 import torch
-model_name = "OpenBMB/UltraRAG-Reranker"
 model = AutoModelForSequenceClassification.from_pretrained(model_name, trust_remote_code=True, torch_dtype=torch.float16).to("cuda")
 # You can use flash-attention 2 to speed up the inference

 from transformers import AutoModelForSequenceClassification
 import torch
+model_name = "OpenBMB/MiniCPM-Reranker-Light"
 model = AutoModelForSequenceClassification.from_pretrained(model_name, trust_remote_code=True, torch_dtype=torch.float16).to("cuda")
 # You can use flash-attention 2 to speed up the inference