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)