theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-03-14 15:57:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #72 from djmaze/patch-1

Browse Source 
Remove explicit install of pytorch & exllamav2 in Dockerfile
This commit is contained in:
Brian Dashore
2024-03-06 01:13:37 -05:00
committed by GitHub
parent 9a007c4707 4afb4137f7
commit 47c42a23d4
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docker/Dockerfile
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