mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-28 15:02:37 +00:00
fix safety test params
This commit is contained in:
parent
3a208f2646
commit
31e7b5ec7f
1 changed files with 2 additions and 2 deletions
4
.github/workflows/integration-tests.yml
vendored
4
.github/workflows/integration-tests.yml
vendored
|
@ -119,8 +119,8 @@ jobs:
|
|||
if [ "${{ matrix.provider }}" == "ollama" ]; then
|
||||
export OLLAMA_URL="http://0.0.0.0:11434"
|
||||
export TEXT_MODEL=ollama/llama3.2:3b-instruct-fp16
|
||||
export SAFETY_MODEL="llama-guard3:1b"
|
||||
EXTRA_PARAMS="--safety-shield=$SAFETY_MODEL"
|
||||
export SAFETY_MODEL="ollama/llama-guard3:1b"
|
||||
EXTRA_PARAMS="--safety-shield=llama-guard"
|
||||
else
|
||||
export VLLM_URL="http://localhost:8000/v1"
|
||||
export TEXT_MODEL=vllm/meta-llama/Llama-3.2-1B-Instruct
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue