bugfix
This commit is contained in:
parent
ff802a1755
commit
33447d53b2
@ -21,7 +21,7 @@ class BaseReranker:
|
|||||||
device = torch.device("mps")
|
device = torch.device("mps")
|
||||||
self.model = self.model.to(device)
|
self.model = self.model.to(device)
|
||||||
|
|
||||||
def process_input(self, pairs):
|
def process_inputs(self, pairs):
|
||||||
inputs = self.tokenizer(
|
inputs = self.tokenizer(
|
||||||
pairs, padding=False, truncation='longest_first',
|
pairs, padding=False, truncation='longest_first',
|
||||||
return_attention_mask=False, max_length=self.max_length
|
return_attention_mask=False, max_length=self.max_length
|
||||||
|
Loading…
Reference in New Issue
Block a user