amazon-chronos v0.1

attila-balint-kul · Apr 25, 2024 · f0ab1d1 · f0ab1d1
1 parent 36b594f
commit f0ab1d1
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/models/amazon-chronos/Dockerfile b/models/amazon-chronos/Dockerfile
@@ -8,8 +8,8 @@ RUN pip install --no-cache-dir -r /usr/local/app/requirements.txt
 COPY ./models /usr/local/app/models
 COPY ./src /usr/local/app/src
 
-ENV MODEL_NAME="chronos-t5-tiny"
-ENV NUM_SAMPLES="20"
+ENV ENFOBENCH_MODEL_NAME="chronos-t5-tiny"
+ENV ENFOBENCH_NUM_SAMPLES="20"
 
 EXPOSE 3000
 CMD ["uvicorn", "src.main:app", "--host", "0.0.0.0", "--port", "3000"]
diff --git a/models/amazon-chronos/src/__init__.py b/models/amazon-chronos/src/__init__.py
diff --git a/models/amazon-chronos/src/main.py b/models/amazon-chronos/src/main.py
@@ -76,8 +76,8 @@ def forecast(
         return forecast
 
 
-model_name = os.getenv("MODEL_NAME")
-num_samples = int(os.getenv("NUM_SAMPLES"))
+model_name = os.getenv("ENFOBENCH_MODEL_NAME")
+num_samples = int(os.getenv("ENFOBENCH_NUM_SAMPLES"))
 
 # Instantiate your model
 model = AmazonChronosModel(model_name=model_name, num_samples=num_samples)