Update benchmarks/auto_tune/auto_tune.sh

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Signed-off-by: Karim Roukoz <roukoz@gmail.com>
This commit is contained in:
Karim Roukoz 2025-12-22 21:15:07 -05:00 committed by GitHub
parent d1b8ff5392
commit 1debdbb01d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -4,6 +4,12 @@
# See details in README (benchmarks/auto_tune/README.md).
TAG=${TAG:-$(date +"%Y_%m_%d_%H_%M")}
# Sanitize TAG to prevent path traversal
if [[ "$TAG" == *..* || "$TAG" == /* ]]; then
echo "Error: TAG cannot contain '..' or be an absolute path." >&2
exit 1
fi
SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
VLLM_LOGGING_LEVEL=${VLLM_LOGGING_LEVEL:-INFO}
BASE=${BASE:-"$SCRIPT_DIR/../../.."}