bugfix
This commit is contained in:
parent
a6d255ffa6
commit
407a1df007
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user