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}