From 5b463a6f798b84dfb4d74cd2544ea436c6c09402 Mon Sep 17 00:00:00 2001
From: Marty Oehme <marty.oehme@gmail.com>
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" ]