mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-22 22:42:25 +00:00
Merge branch 'main' into fix-2729-vector-io-embedding-config
This commit is contained in:
commit
1c7405384d
2 changed files with 6 additions and 6 deletions
|
|
@ -262,6 +262,11 @@ inference_store:
|
||||||
type: sqlite
|
type: sqlite
|
||||||
db_path: ${env.SQLITE_STORE_DIR:=~/.llama/distributions/starter}/inference_store.db
|
db_path: ${env.SQLITE_STORE_DIR:=~/.llama/distributions/starter}/inference_store.db
|
||||||
models:
|
models:
|
||||||
|
- metadata:
|
||||||
|
embedding_dimension: 384
|
||||||
|
model_id: all-MiniLM-L6-v2
|
||||||
|
provider_id: ${env.ENABLE_SENTENCE_TRANSFORMERS:=sentence-transformers}
|
||||||
|
model_type: embedding
|
||||||
- metadata: {}
|
- metadata: {}
|
||||||
model_id: ${env.ENABLE_CEREBRAS:=__disabled__}/llama3.1-8b
|
model_id: ${env.ENABLE_CEREBRAS:=__disabled__}/llama3.1-8b
|
||||||
provider_id: ${env.ENABLE_CEREBRAS:=__disabled__}
|
provider_id: ${env.ENABLE_CEREBRAS:=__disabled__}
|
||||||
|
|
@ -1168,11 +1173,6 @@ models:
|
||||||
provider_id: ${env.ENABLE_SAMBANOVA:=__disabled__}
|
provider_id: ${env.ENABLE_SAMBANOVA:=__disabled__}
|
||||||
provider_model_id: sambanova/Meta-Llama-Guard-3-8B
|
provider_model_id: sambanova/Meta-Llama-Guard-3-8B
|
||||||
model_type: llm
|
model_type: llm
|
||||||
- metadata:
|
|
||||||
embedding_dimension: 384
|
|
||||||
model_id: all-MiniLM-L6-v2
|
|
||||||
provider_id: ${env.ENABLE_SENTENCE_TRANSFORMERS:=sentence-transformers}
|
|
||||||
model_type: embedding
|
|
||||||
shields:
|
shields:
|
||||||
- shield_id: ${env.SAFETY_MODEL:=__disabled__}
|
- shield_id: ${env.SAFETY_MODEL:=__disabled__}
|
||||||
provider_shield_id: ${env.ENABLE_OLLAMA:=__disabled__}/${env.SAFETY_MODEL:=__disabled__}
|
provider_shield_id: ${env.ENABLE_OLLAMA:=__disabled__}/${env.SAFETY_MODEL:=__disabled__}
|
||||||
|
|
|
||||||
|
|
@ -323,7 +323,7 @@ def get_distribution_template() -> DistributionTemplate:
|
||||||
"files": [files_provider],
|
"files": [files_provider],
|
||||||
"post_training": [post_training_provider],
|
"post_training": [post_training_provider],
|
||||||
},
|
},
|
||||||
default_models=default_models + [embedding_model],
|
default_models=[embedding_model] + default_models,
|
||||||
default_tool_groups=default_tool_groups,
|
default_tool_groups=default_tool_groups,
|
||||||
# TODO: add a way to enable/disable shields on the fly
|
# TODO: add a way to enable/disable shields on the fly
|
||||||
default_shields=shields,
|
default_shields=shields,
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue