Spaces:
Runtime error
Runtime error
Update apis/chat_api.py
Browse files- apis/chat_api.py +2 -2
apis/chat_api.py
CHANGED
|
@@ -49,9 +49,9 @@ class EmbeddingsAPIInference:
|
|
| 49 |
|
| 50 |
|
| 51 |
class SentenceTransformerLocal(SentenceTransformer):
|
| 52 |
-
def encode(self, **kwargs):
|
| 53 |
kwargs.pop("api_key", None)
|
| 54 |
-
return super().encode(**kwargs).tolist()
|
| 55 |
|
| 56 |
|
| 57 |
|
|
|
|
| 49 |
|
| 50 |
|
| 51 |
class SentenceTransformerLocal(SentenceTransformer):
|
| 52 |
+
def encode(self, *args, **kwargs):
|
| 53 |
kwargs.pop("api_key", None)
|
| 54 |
+
return super().encode(*args, **kwargs).tolist()
|
| 55 |
|
| 56 |
|
| 57 |
|