From 5b463a6f798b84dfb4d74cd2544ea436c6c09402 Mon Sep 17 00:00:00 2001 From: Marty Oehme Date: Mon, 21 Aug 2023 18:51:23 +0200 Subject: [PATCH] Only run light-the-torch when installed in env --- Dockerfile | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index c6f670c..8f44143 100644 --- a/Dockerfile +++ b/Dockerfile @@ -24,8 +24,6 @@ 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 ./ @@ -45,6 +43,9 @@ WORKDIR ${APP_PATH} RUN poetry install RUN mkdir -p ${VERBANOTE_OUTPUT_PATH} +# installing the large models +RUN poetry run ltt install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 + COPY ./docker/docker-entrypoint.sh /docker-entrypoint.sh RUN chmod +x /docker-entrypoint.sh ENTRYPOINT [ "/docker-entrypoint.sh" ]