feat: comfyui dockerfile

2023-12-27 12:13:48 +08:00 · 2023-12-27 12:13:48 +08:00 · f24b95fce7
parent 0eecbd1fd4
commit f24b95fce7
1 changed files with 2 additions and 2 deletions
--- a/build_scripts/Dockerfile.comfyui.from_scratch
+++ b/build_scripts/Dockerfile.comfyui.from_scratch
@ -27,5 +27,5 @@ RUN echo "/opt/ml/code" > "/opt/conda/lib/python3.10/site-packages/packages.pth"
 WORKDIR /opt/ml/code

 ENV ON_DOCKER true
-
-ENTRYPOINT ["python", "/opt/ml/code/main.py"]
+ENV SAGEMAKER_PROGRAM /opt/ml/code/main.py
+#ENTRYPOINT ["python", "/opt/ml/code/main.py"]