From f75b16dd594a88466ce5cf91d2c91581a5f4825f Mon Sep 17 00:00:00 2001
From: attilabalint <balint.attil@gmail.com>
Date: Thu, 13 Jun 2024 11:46:27 +0200
Subject: [PATCH] separated amazon models into individual containers

---
 models/amazon-chronos/Dockerfile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/models/amazon-chronos/Dockerfile b/models/amazon-chronos/Dockerfile
index 45c202f..b3edfbb 100644
--- 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