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

fix config w/ safety

Browse source
This commit is contained in:
Xi Yan 2024-09-22 23:27:04 -07:00
parent 4586692dee
commit 1ac188e1b3
4 changed files with 28 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

29
tests/examples/router-local-run.yaml
View file

@ -38,11 +38,30 @@ routing_table:
- routing_key: llama_guard
provider_id: meta-reference
config:
model: Llama-Guard-3-8B
excluded_categories: []
disable_input_check: false
disable_output_check: false
llama_guard_shield:
model: Llama-Guard-3-8B
excluded_categories: []
disable_input_check: false
disable_output_check: false
prompt_guard_shield:
model: Prompt-Guard-86M
- routing_key: prompt_guard
provider_id: meta-reference
config:
model: Prompt-Guard-86M
llama_guard_shield:
model: Llama-Guard-3-8B
excluded_categories: []
disable_input_check: false
disable_output_check: false
prompt_guard_shield:
model: Prompt-Guard-86M
- routing_key: injection_shield
provider_id: meta-reference
config:
llama_guard_shield:
model: Llama-Guard-3-8B
excluded_categories: []
disable_input_check: false
disable_output_check: false
prompt_guard_shield:
model: Prompt-Guard-86M