phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 13:00:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix embedding metadata: embedding_dimensions -> embedding_dimension

Browse source
This commit is contained in:
Matthew Farrellee 2025-02-21 08:02:23 -06:00
parent 34226d6c93
commit a37dca7e02
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/inference/nvidia/models.py
View file

@ -52,7 +52,7 @@ _MODEL_ENTRIES = [
provider_model_id="baai/bge-m3",
model_type=ModelType.embedding,
metadata={
"embedding_dimensions": 1024,
"embedding_dimension": 1024,
"context_length": 8192,
},
),

2
llama_stack/templates/nvidia/run.yaml
View file

@ -136,7 +136,7 @@ models:
provider_model_id: meta/llama-3.2-90b-vision-instruct
model_type: llm
- metadata:
embedding_dimensions: 1024
embedding_dimension: 1024
context_length: 8192
model_id: baai/bge-m3
provider_id: nvidia