Remove explicit pytorch & exllamav2 in Dockerfile

These packages are already installed via requirements.txt.
This commit is contained in:
Martin Honermeyer
2024-02-25 18:03:01 +01:00
committed by GitHub
parent fc857893ee
commit 4afb4137f7

View File

@@ -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