bugfix
This commit is contained in:
parent
a6d255ffa6
commit
407a1df007
@ -25,6 +25,7 @@ class BgeReranker(BaseReranker):
|
|||||||
def compute_logits(self, inputs):
|
def compute_logits(self, inputs):
|
||||||
scores = self.model(**inputs,
|
scores = self.model(**inputs,
|
||||||
return_dict=True).logits.view(-1, ).float()
|
return_dict=True).logits.view(-1, ).float()
|
||||||
|
scores = [ s.item() for s in scores ]
|
||||||
return scores
|
return scores
|
||||||
|
|
||||||
llm_register('bge-reranker', BgeReranker)
|
llm_register('bge-reranker', BgeReranker)
|
||||||
|
Loading…
Reference in New Issue
Block a user