From 6e72776fbf47740cc4ad36a659e74046b9941d49 Mon Sep 17 00:00:00 2001 From: Marty Oehme Date: Sun, 20 Aug 2023 13:28:31 +0200 Subject: [PATCH] Install torch models in Dockerfile --- Dockerfile | 3 +++ 1 file changed, 3 insertions(+) diff --git a/Dockerfile b/Dockerfile index 1222f9a..c2b1922 100644 --- a/Dockerfile +++ b/Dockerfile @@ -24,6 +24,9 @@ ENV \ RUN curl -sSL https://install.python-poetry.org | python3 - ENV PATH="${POETRY_HOME}/bin:${PATH}" +# installing the large models +RUN ltt install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 + WORKDIR ${APP_PATH} COPY ./poetry.lock ./pyproject.toml ./ COPY ./${APP_NAME} ./${APP_NAME}