From 0e4870b6e85aedc7a5bf0585cf3471a28bffaa44 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Sun, 23 Apr 2023 16:21:38 +0800
Subject: [PATCH] Fix cuda issue

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 s_bert.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/s_bert.py b/s_bert.py
index 77f5dd1..e431525 100644
--- a/s_bert.py
+++ b/s_bert.py
@@ -89,8 +89,8 @@ class STransformers(NNOperator):
         else:
             sentences = txt
         inputs = self.tokenize(sentences)
-        for k, v in inputs.items():
-            inputs[k] = v.to(self.device)
+#         for k, v in inputs.items():
+#             inputs[k] = v.to(self.device)
         embs = self.model(**inputs).cpu().detach().numpy()
         if isinstance(txt, str):
             embs = embs.squeeze(0)