forked from phoenix-oss/llama-stack-mirror
c1f7ba3aed
Splits the meta-reference safety implementation into three distinct providers: - inline::llama-guard - inline::prompt-guard - inline::code-scanner Note that this PR is a backward incompatible change to the llama stack server. I have added deprecation_error field to ProviderSpec -- the server reads it and immediately barfs. This is used to direct the user with a specific message on what action to perform. An automagical "config upgrade" is a bit too much work to implement right now :/ (Note that we will be gradually prefixing all inline providers with inline:: -- I am only doing this for this set of new providers because otherwise existing configuration files will break even more badly.)
59 lines
1.3 KiB
YAML
59 lines
1.3 KiB
YAML
version: '2'
|
|
built_at: '2024-10-08T17:40:45.325529'
|
|
image_name: local
|
|
docker_image: null
|
|
conda_env: local
|
|
apis:
|
|
- shields
|
|
- agents
|
|
- models
|
|
- memory
|
|
- memory_banks
|
|
- inference
|
|
- safety
|
|
providers:
|
|
inference:
|
|
- provider_id: meta0
|
|
provider_type: meta-reference-quantized
|
|
config:
|
|
model: Llama3.2-3B-Instruct:int4-qlora-eo8
|
|
quantization:
|
|
type: int4
|
|
torch_seed: null
|
|
max_seq_len: 2048
|
|
max_batch_size: 1
|
|
- provider_id: meta1
|
|
provider_type: meta-reference-quantized
|
|
config:
|
|
# not a quantized model !
|
|
model: Llama-Guard-3-1B
|
|
quantization: null
|
|
torch_seed: null
|
|
max_seq_len: 2048
|
|
max_batch_size: 1
|
|
safety:
|
|
- provider_id: meta0
|
|
provider_type: inline::llama-guard
|
|
config:
|
|
model: Llama-Guard-3-1B
|
|
excluded_categories: []
|
|
- provider_id: meta1
|
|
provider_type: inline::prompt-guard
|
|
config:
|
|
model: Prompt-Guard-86M
|
|
memory:
|
|
- provider_id: meta0
|
|
provider_type: meta-reference
|
|
config: {}
|
|
agents:
|
|
- provider_id: meta0
|
|
provider_type: meta-reference
|
|
config:
|
|
persistence_store:
|
|
namespace: null
|
|
type: sqlite
|
|
db_path: ~/.llama/runtime/kvstore.db
|
|
telemetry:
|
|
- provider_id: meta0
|
|
provider_type: meta-reference
|
|
config: {}
|