Fix Docker deployment

.dockerignore

@@ -0,0 +1,17 @@
+.git
+__pycache__
+*.pyc
+*.pyo
+*.egg-info
+build/
+dist/
+.env
+.venv
+env/
+venv/
+tests/output/
+tests/merged/
+logs/
+.mypy_cache/
+.pytest_cache/
+.ruff_cache/

.github/workflows/docker.yml

@@ -4,12 +4,14 @@ on:
   push:
     branches:
       - main
+      # - fix/docker-deployment 
   pull_request:
     branches:
       - main
+  workflow_dispatch: 
 
 env:
-  IMAGE_NAME: michelducartier24/mirage
+  IMAGE_NAME: fabnemepfl/mirage
   REGISTRY: docker.io
 
 jobs:
@@ -62,6 +64,7 @@ jobs:
         uses: actions/checkout@v4
 
       - name: Log in to DockerHub
+        if: github.event_name != 'pull_request'
         uses: docker/login-action@v3
         with:
           username: ${{ secrets.DOCKER_USERNAME }}
@@ -75,7 +78,7 @@ jobs:
         with:
           context: .
           file: ${{ matrix.path }}
-          push: true
+          push: ${{ github.event_name != 'pull_request' }}
           tags: |
             ${{ secrets.DOCKER_USERNAME }}/${{ matrix.name }}:latest-${{ matrix.tag_base }}
-            ${{ secrets.DOCKER_USERNAME }}/${{ matrix.name }}:${{ github.sha }}-${{ matrix.tag_base }}
+            ${{ secrets.DOCKER_USERNAME }}/${{ matrix.name }}:${{ github.sha }}-${{ matrix.tag_base }}

README.md

@@ -16,6 +16,27 @@ For testing and scripts that make use of the library, it is advised to create a
 ./scripts/generate_env.sh
 ```
 
+## Docker 
+
+### Build
+
+```bash
+docker compose build
+```
+
+### Run
+
+```bash
+docker compose run --rm mmirage --config configs/your_config.yaml
+```
+
+The container requires an NVIDIA GPU. The `docker-compose.yml` is configured to request GPU access, but the host must have:
+- NVIDIA GPU drivers installed
+- NVIDIA Container Toolkit / `nvidia-container-runtime` configured for Docker
+- A recent Docker Engine and Docker Compose version with GPU support enabled
+
+Without these host-side prerequisites, `docker compose run` may fail to detect or use the GPU.
+
 ## Key features
 
 - **Multimodal Support**: Process both text and images with vision-language models

docker-compose.yml

@@ -0,0 +1,12 @@
+services:
+  mmirage:
+    build:
+      context: .
+      dockerfile: docker/Dockerfile
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]

docker/Dockerfile

@@ -1,5 +1,11 @@
-FROM docker.io/lmsysorg/sglang:latest
+FROM --platform=linux/amd64 lmsysorg/sglang:v0.5.8.post1-runtime
+
+ENV NVIDIA_VISIBLE_DEVICES=all
+ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
 
 COPY . /workspace/MMIRAGE
 WORKDIR /workspace/MMIRAGE
-RUN pip install --no-cache-dir -e .
+
+RUN pip install --no-cache-dir .
+
+ENTRYPOINT ["python3", "-c", "import mmirage.core.process, runpy; runpy.run_module('mmirage.shard_process', run_name='__main__')"]