diff --git a/Dockerfile.cpu b/Dockerfile.cpu index 8cc83e6..fbbb1b3 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -11,7 +11,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ # Copy and install dependencies COPY requirements.txt . -RUN pip install --no-cache-dir --prefix=/install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cpu && pip install --no-cache-dir --prefix=/install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu +RUN pip install --no-cache-dir --prefix=/install -r requirements.txt # === Stage 2: Create a minimal runtime image === FROM python:3.11-slim-bullseye AS runtime