From 48e0ba13418f690d338e5111d5d432baef1f1724 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Mon, 13 Mar 2023 17:15:42 +0800 Subject: [PATCH] Fix indent issue Signed-off-by: Jael Gu --- s_bert.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/s_bert.py b/s_bert.py index 1a73b58..7586c2e 100644 --- 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)