phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Kill noise from test output

Browse source
This commit is contained in:
Ashwin Bharambe 2025-02-21 15:37:23 -08:00
parent bf38d0aba0
commit 45ffe87d7c
3 changed files with 4 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/utils/inference/embedding_mixin.py
View file

@ -35,7 +35,9 @@ class SentenceTransformerEmbeddingMixin:
) -> EmbeddingsResponse:
model = await self.model_store.get_model(model_id)
embedding_model = self._load_sentence_transformer_model(model.provider_resource_id)
embeddings = embedding_model.encode([interleaved_content_as_str(content) for content in contents])
embeddings = embedding_model.encode(
[interleaved_content_as_str(content) for content in contents], show_progress_bar=False
)
return EmbeddingsResponse(embeddings=embeddings)
def _load_sentence_transformer_model(self, model: str) -> "SentenceTransformer":