moved main.py run to entrypoint.sh

cyntachs · Aug 11, 2023 · 0b19482 · 0b19482
1 parent cf79ba0
commit 0b19482
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -9,6 +9,4 @@ ENV ROOT=/stable-diffusion
 RUN mkdir ${ROOT}
 
 EXPOSE 5555
-ENTRYPOINT [ "sh" , "entrypoint.sh" ]
-WORKDIR ${ROOT}
-CMD python -u main.py --listen --port 5555 ${CLI_ARGS}
+ENTRYPOINT [ "sh" , "entrypoint.sh" ]
diff --git a/entrypoint.sh b/entrypoint.sh
@@ -1,5 +1,7 @@
 #!/bin/bash
 
+cd /
+
 if [ ! -f "/stable-diffusion/.installed" ]; then
     echo "Installing ComfyUI..."
     git clone https://github.com/comfyanonymous/ComfyUI.git /stable-diffusion
@@ -21,10 +23,13 @@ if [ ! -f "/stable-diffusion/.installed" ]; then
     git submodule update --init --recursive
     python install.py
 
-    cd /stable-diffusion
+    cd /
 
     touch /stable-diffusion/.installed
 else
     chmod +x /update.sh
     sh /update.sh
 fi;
+
+cd /stable-diffusion
+python -u main.py --listen --port 5555 ${CLI_ARGS}
diff --git a/update.sh b/update.sh
@@ -12,4 +12,6 @@ cd ./custom_nodes/ComfyUI-Impact-Pack
 git fetch https://github.com/ltdrdata/ComfyUI-Impact-Pack.git
 git pull
 
+cd /
+
 echo "Done."