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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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