bugfix
This commit is contained in:
parent
c31199d2e3
commit
a6d255ffa6
@ -18,7 +18,7 @@ class BgeReranker(BaseReranker):
|
||||
return [[query, doc] for doc in docs]
|
||||
|
||||
def process_inputs(self, pairs):
|
||||
inputs = tokenizer(pairs, padding=True,
|
||||
inputs = self.tokenizer(pairs, padding=True,
|
||||
truncation=True, return_tensors='pt', max_length=512)
|
||||
return inputs
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user