init

Dockerfile.worker

@@ -0,0 +1,75 @@
+# Dockerfile for Distributed Llama Worker (Raspberry Pi)
+# This variant runs as a worker node and connects to a controller
+FROM arm64v8/debian:bookworm-slim
+
+# Install dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    g++ \
+    make \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set working directory
+WORKDIR /app
+
+# Copy source code
+COPY src/ ./src/
+COPY Makefile ./
+
+# 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
+
+# Use the entrypoint script
+ENTRYPOINT ["/app/entrypoint.sh"]