This commit is contained in:
yumoqing 2025-06-21 11:43:11 +08:00
parent c31199d2e3
commit a6d255ffa6

View File

@ -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