Update Dockerfile

please work
actuarialopensource · May 3, 2024 · 4b6fd49 · 4b6fd49
1 parent ad5f874
commit 4b6fd49
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/containers/BasicTerm_ME_heavylight/Dockerfile b/containers/BasicTerm_ME_heavylight/Dockerfile
@@ -16,5 +16,4 @@ COPY . /app/
 ENV PYTORCH_CUDA_ALLOC_CONF="garbage_collection_threshold:0.8"
 
 # Set the entrypoint and provide the script name as default command
-ENTRYPOINT ["python"]
-CMD ["main.py"]
+ENTRYPOINT ["python", "main.py"]