From a07c49e4efe09c08d3a2c1d103ff9eb7ce34913f Mon Sep 17 00:00:00 2001
From: ChengZi <chen.zhang@zilliz.com>
Date: Wed, 26 Jul 2023 11:04:34 +0800
Subject: [PATCH] remove redundant to_device

Signed-off-by: ChengZi <chen.zhang@zilliz.com>
---
 rerank.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/rerank.py b/rerank.py
index 510e81a..4840319 100644
--- 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)