From 4afb4137f799e97c87ec82485d9454e233752bd4 Mon Sep 17 00:00:00 2001 From: Martin Honermeyer <7229+djmaze@users.noreply.github.com> Date: Sun, 25 Feb 2024 18:03:01 +0100 Subject: [PATCH] Remove explicit pytorch & exllamav2 in Dockerfile These packages are already installed via requirements.txt. --- docker/Dockerfile | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 5c7e46bb..ded61b52 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -13,11 +13,7 @@ WORKDIR /usr/src/app # Get requirements COPY requirements.txt requirements.txt -# Install torch with CUDA support and exllamav2 -RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu121 -RUN pip install exllamav2 - -# Install any other needed packages specified in requirements.txt +# Install packages specified in requirements.txt RUN pip install --no-cache-dir -r requirements.txt # Copy the current directory contents into the container at /usr/src/app