new stuff
This commit is contained in:
@@ -16,58 +16,15 @@ WORKDIR /app
|
||||
COPY src/ ./src/
|
||||
COPY Makefile ./
|
||||
|
||||
# Copy worker entrypoint script
|
||||
COPY scripts/entrypoint-worker.sh /app/entrypoint.sh
|
||||
|
||||
# Make script executable
|
||||
RUN chmod +x /app/entrypoint.sh
|
||||
|
||||
# Build only the worker application
|
||||
RUN make dllama
|
||||
|
||||
# Create entrypoint script
|
||||
COPY <<EOF /app/entrypoint.sh
|
||||
#!/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
|
||||
EOF
|
||||
|
||||
RUN chmod +x /app/entrypoint.sh
|
||||
|
||||
# Expose the default worker port
|
||||
EXPOSE 9999
|
||||
|
||||
|
||||
Reference in New Issue
Block a user