This commit is contained in:
IAN MILLER 2025-10-03 04:57:49 -07:00 committed by GitHub
commit 7774d9e9f3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
80 changed files with 32197 additions and 186 deletions

View file

@ -86,7 +86,7 @@ class SentenceTransformerEmbeddingMixin:
def _load_model():
from sentence_transformers import SentenceTransformer
return SentenceTransformer(model)
return SentenceTransformer(model, trust_remote_code=True)
loaded_model = await asyncio.to_thread(_load_model)
EMBEDDING_MODELS[model] = loaded_model