feat: add clip_hg executor

jina-ai · Jun 1, 2022 · 5f7a115 · 5f7a115
1 parent a3dd39b
commit 5f7a115
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/clip_server/executors/clip_hg.py b/server/clip_server/executors/clip_hg.py
@@ -194,4 +194,4 @@ def _tokenize_texts(self, texts: Sequence[str]):
             truncation=True,
             return_tensors='pt',
         )
-        return {k: v.to(self.device) for k, v in x.items()}
+        return {k: v.to(self.device) for k, v in x.items()}