Kokoro-FastAPI/test_client/start.sh
remsky e5e85b32d2 -Add model instance pooling, better concurrency
-Add load testing setup with Locust
2025-02-09 23:03:16 -07:00

16 lines
No EOL
469 B
Bash

#!/bin/bash
# If LOCUST_HEADLESS is true, run in headless mode with specified parameters
if [ "$LOCUST_HEADLESS" = "true" ]; then
locust -f locustfile.py \
--host ${LOCUST_HOST} \
--users ${LOCUST_USERS} \
--spawn-rate ${LOCUST_SPAWN_RATE} \
--run-time ${LOCUST_RUN_TIME} \
--headless \
--print-stats \
--html report.html
else
# Run with web interface
locust -f locustfile.py --host ${LOCUST_HOST}
fi