This commit is contained in:
yumoqing 2025-06-21 11:24:53 +08:00
parent ff802a1755
commit 33447d53b2

View File

@ -21,7 +21,7 @@ class BaseReranker:
device = torch.device("mps")
self.model = self.model.to(device)
def process_input(self, pairs):
def process_inputs(self, pairs):
inputs = self.tokenizer(
pairs, padding=False, truncation='longest_first',
return_attention_mask=False, max_length=self.max_length