hf-llm-api

Sleeping

jonathanjordan21 commited on Jul 27

Commit

6f4d127

•

1 Parent(s): 827aa76

Update apis/chat_api.py

Files changed (1) hide show

apis/chat_api.py CHANGED Viewed

@@ -49,9 +49,9 @@ class EmbeddingsAPIInference:
 class SentenceTransformerLocal(SentenceTransformer):
-    def encode(self, **kwargs):
         kwargs.pop("api_key", None)
-        return super().encode(**kwargs).tolist()

 class SentenceTransformerLocal(SentenceTransformer):
+    def encode(self, *args, **kwargs):
         kwargs.pop("api_key", None)
+        return super().encode(*args, **kwargs).tolist()