mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-13 04:22:35 +00:00
refactor: switch to the new default nomic-embed-text-v1.5 embedding model in LS
This commit is contained in:
parent
b95f095a54
commit
429f1d2405
51 changed files with 16149 additions and 83 deletions
|
|
@ -83,7 +83,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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue