This commit is contained in:
yumoqing 2025-06-21 11:53:08 +08:00
parent a6d255ffa6
commit 407a1df007

View File

@ -25,6 +25,7 @@ class BgeReranker(BaseReranker):
def compute_logits(self, inputs):
scores = self.model(**inputs,
return_dict=True).logits.view(-1, ).float()
scores = [ s.item() for s in scores ]
return scores
llm_register('bge-reranker', BgeReranker)