From 31e7b5ec7f004187945d1e73645488d0ea44ea74 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Fri, 25 Jul 2025 14:44:05 -0700 Subject: [PATCH] fix safety test params --- .github/workflows/integration-tests.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml index 62352520f..be2613fbb 100644 --- a/.github/workflows/integration-tests.yml +++ b/.github/workflows/integration-tests.yml @@ -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