dockerfile

sweepai · Aug 29, 2023 · 352c0c6 · 352c0c6
1 parent 25b83bd
commit 352c0c6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -11,6 +11,8 @@ RUN apt-get update \
     && rm -rf /var/lib/apt/lists/*
 
 COPY pyproject.toml ./
+
+RUN pip install torch --index-url https://download.pytorch.org/whl/cpu
 RUN pip install sentence_transformers --no-cache-dir
 RUN pip install --no-cache-dir poetry \
     && poetry export -f requirements.txt --without-hashes -o requirements.txt \