From 33447d53b2c09bb3b914dfcfc0d1e4d46faa1380 Mon Sep 17 00:00:00 2001 From: yumoqing Date: Sat, 21 Jun 2025 11:24:53 +0800 Subject: [PATCH] bugfix --- llmengine/base_reranker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmengine/base_reranker.py b/llmengine/base_reranker.py index 39a0570..268327b 100644 --- a/llmengine/base_reranker.py +++ b/llmengine/base_reranker.py @@ -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