Browse Source

fix: dockerfile

Timothy J. Baek 1 year ago
parent
commit
02f364bfd9
2 changed files with 2 additions and 7 deletions
  1. 2 0
      Dockerfile
  2. 0 7
      backend/requirements.txt

+ 2 - 0
Dockerfile

@@ -40,6 +40,8 @@ COPY --from=build /app/build /app/build
 WORKDIR /app/backend
 
 COPY ./backend/requirements.txt ./requirements.txt
+
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
 RUN pip3 install -r requirements.txt
 
 # RUN python -c "from sentence_transformers import SentenceTransformer; model = SentenceTransformer('all-MiniLM-L6-v2')"

+ 0 - 7
backend/requirements.txt

@@ -16,13 +16,6 @@ aiohttp
 peewee
 bcrypt
 
---index-url https://download.pytorch.org/whl/cpu
-torch 
---index-url https://download.pytorch.org/whl/cpu
-torchvision 
---index-url https://download.pytorch.org/whl/cpu
-torchaudio 
-
 langchain
 langchain-community
 chromadb