From 02f364bfd9501834c70c834c9d551e8a617df656 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Sun, 7 Jan 2024 21:22:37 -0800 Subject: [PATCH] fix: dockerfile --- Dockerfile | 2 ++ backend/requirements.txt | 7 ------- 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/Dockerfile b/Dockerfile index 8e81b774..64a30753 100644 --- a/Dockerfile +++ b/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')" diff --git a/backend/requirements.txt b/backend/requirements.txt index e0c2f25a..06af0efc 100644 --- a/backend/requirements.txt +++ b/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