remove redundant to_device

Signed-off-by: ChengZi <chen.zhang@zilliz.com>
2 years ago · a07c49e4ef
1 changed files with 0 additions and 3 deletions
--- a/rerank.py
+++ b/rerank.py
@ -65,9 +65,6 @@ class ReRank(NNOperator):

        tokenized = self.tokenizer(*texts, padding=True, truncation='longest_first', return_tensors="pt", max_length=self.max_length)

-        for name in tokenized:
-            tokenized[name] = tokenized[name].to(self.device)
-
        logits = self.model(**tokenized)
        scores = self.post_proc(logits)