fix: dockerfile

This commit is contained in:
Timothy J. Baek 2024-01-07 21:22:37 -08:00
parent 706a8d4f8d
commit 02f364bfd9
2 changed files with 2 additions and 7 deletions

View file

@ -40,6 +40,8 @@ COPY --from=build /app/build /app/build
WORKDIR /app/backend WORKDIR /app/backend
COPY ./backend/requirements.txt ./requirements.txt 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 pip3 install -r requirements.txt
# RUN python -c "from sentence_transformers import SentenceTransformer; model = SentenceTransformer('all-MiniLM-L6-v2')" # RUN python -c "from sentence_transformers import SentenceTransformer; model = SentenceTransformer('all-MiniLM-L6-v2')"

View file

@ -16,13 +16,6 @@ aiohttp
peewee peewee
bcrypt 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
langchain-community langchain-community
chromadb chromadb