bugfix
This commit is contained in:
parent
ff802a1755
commit
33447d53b2
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user