43 lines
971 B
Bash
43 lines
971 B
Bash
#!/bin/bash
|
|
|
|
# Default values
|
|
PORT=9999
|
|
NTHREADS=4
|
|
|
|
# Parse command line arguments
|
|
while [[ $# -gt 0 ]]; do
|
|
case $1 in
|
|
--port)
|
|
PORT="$2"
|
|
shift 2
|
|
;;
|
|
--nthreads)
|
|
NTHREADS="$2"
|
|
shift 2
|
|
;;
|
|
--help)
|
|
echo "Usage: docker run distributed-llama-worker [OPTIONS]"
|
|
echo ""
|
|
echo "Options:"
|
|
echo " --port <port> Worker port (default: 9999)"
|
|
echo " --nthreads <n> Number of threads (default: 4)"
|
|
echo ""
|
|
echo "Example:"
|
|
echo " docker run -p 9999:9999 distributed-llama-worker --port 9999 --nthreads 4"
|
|
exit 0
|
|
;;
|
|
*)
|
|
echo "Unknown option: $1"
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# Build the command
|
|
CMD="./dllama worker --port $PORT --nthreads $NTHREADS"
|
|
|
|
echo "Starting worker with command:"
|
|
echo "$CMD"
|
|
echo ""
|
|
|
|
exec $CMD |