separated amazon models into individual containers

attila-balint-kul · Jun 13, 2024 · f75b16d · f75b16d
1 parent 41633de
commit f75b16d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/models/amazon-chronos/Dockerfile b/models/amazon-chronos/Dockerfile
@@ -5,10 +5,11 @@ WORKDIR /usr/local/app
 COPY ./requirements.txt /usr/local/app/requirements.txt
 RUN pip install --no-cache-dir -r /usr/local/app/requirements.txt
 
-COPY ./models /usr/local/app/models
+ARG MODEL_SIZE
+COPY ./models/chronos-t5-${MODEL_SIZE} /usr/local/app/models/chronos-t5-${MODEL_SIZE}
 COPY ./src /usr/local/app/src
 
-ENV ENFOBENCH_MODEL_NAME="chronos-t5-tiny"
+ENV ENFOBENCH_MODEL_NAME="chronos-t5-${MODEL_SIZE}"
 ENV ENFOBENCH_NUM_SAMPLES="1"
 
 EXPOSE 3000