From e6f3f01947f1a00b03ee13dac2d664805acf91db Mon Sep 17 00:00:00 2001 From: philschmid Date: Fri, 21 Oct 2022 09:40:06 +0000 Subject: [PATCH] fix in base --- dockerfiles/starlette/pytorch/Dockerfile.cpu | 1 + dockerfiles/starlette/pytorch/Dockerfile.gpu | 1 + 2 files changed, 2 insertions(+) diff --git a/dockerfiles/starlette/pytorch/Dockerfile.cpu b/dockerfiles/starlette/pytorch/Dockerfile.cpu index 03c0d9a6..53f51df7 100644 --- a/dockerfiles/starlette/pytorch/Dockerfile.cpu +++ b/dockerfiles/starlette/pytorch/Dockerfile.cpu @@ -3,6 +3,7 @@ FROM huggingface/transformers-inference:4.23.1-pt1.11-cpu # install starlette framework COPY starlette_requirements.txt /tmp/requirements.txt RUN pip install --no-cache-dir -r /tmp/requirements.txt && rm /tmp/requirements.txt + # Think about a better solution -> base contaienr has pt 1.11. thats why need below 0.13 RUN pip install --no-cache-dir sentence_transformers torchvision~="0.12.0" diff --git a/dockerfiles/starlette/pytorch/Dockerfile.gpu b/dockerfiles/starlette/pytorch/Dockerfile.gpu index 69ab5de3..cd8bc8dd 100644 --- a/dockerfiles/starlette/pytorch/Dockerfile.gpu +++ b/dockerfiles/starlette/pytorch/Dockerfile.gpu @@ -3,6 +3,7 @@ FROM huggingface/transformers-inference:4.23.1-pt1.11-cuda11.5 # install starlette framework COPY starlette_requirements.txt /tmp/requirements.txt RUN pip install --no-cache-dir -r /tmp/requirements.txt && rm /tmp/requirements.txt + # Think about a better solution -> base contaienr has pt 1.11. thats why need below 0.13 RUN pip install --no-cache-dir sentence_transformers torchvision~="0.12.0"