Fix indent issue

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 48e0ba1341
1 changed files with 2 additions and 2 deletions
--- a/s_bert.py
+++ b/s_bert.py
@ -85,8 +85,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)