Updated start scripts

ashleykleynhans · Jul 3, 2024 · a3474cf · a3474cf
1 parent 907946d
commit a3474cf
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 4 deletions.
diff --git a/scripts/start_a1111.sh b/scripts/start_a1111.sh
@@ -1,2 +1,4 @@
 #!/usr/bin/env bash
-cd /workspace/stable-diffusion-webui && nohup ./webui.sh -f > /workspace/logs/webui.log 2>&1 &
+export PYTHONUNBUFFERED=1
+cd /workspace/stable-diffusion-webui
+nohup ./webui.sh -f > /workspace/logs/webui.log 2>&1 &
diff --git a/scripts/start_comfyui.sh b/scripts/start_comfyui.sh
@@ -1,2 +1,8 @@
 #!/usr/bin/env bash
-cd /workspace/ComfyUI && source venv/bin/activate && python3 main.py --listen 0.0.0.0 --port 3021 > /workspace/logs/comfyui.log 2>&1 &
+export PYTHONUNBUFFERED=1
+cd /workspace/ComfyUI
+source venv/bin/activate
+TCMALLOC="$(ldconfig -p | grep -Po "libtcmalloc.so.\d" | head -n 1)"
+export LD_PRELOAD="${TCMALLOC}"
+python3 main.py --listen 0.0.0.0 --port 3021 > /workspace/logs/comfyui.log 2>&1 &
+deactivate
diff --git a/scripts/start_invokeai.sh b/scripts/start_invokeai.sh
@@ -1,2 +1,5 @@
 #!/usr/bin/env bash
-cd /workspace/InvokeAI && source venv/bin/activate && nohup invokeai-web > /workspace/logs/invokeai.log 2>&1 &
+export PYTHONUNBUFFERED=1
+cd /workspace/InvokeAI
+source /venvs/invokeai/bin/activate
+nohup invokeai-web > /workspace/logs/invokeai.log 2>&1 &
diff --git a/scripts/start_kohya.sh b/scripts/start_kohya.sh
@@ -1,2 +1,6 @@
 #!/usr/bin/env bash
-cd /workspace/kohya_ss && nohup ./gui.sh --listen 0.0.0.0 --server_port 3011 --headless > /workspace/logs/kohya_ss.log 2>&1 &
+export PYTHONUNBUFFERED=1
+cd /workspace/kohya_ss
+export HF_HOME="/workspace"
+source /venvs/kohya_ss/bin/activate
+nohup ./gui.sh --listen 0.0.0.0 --server_port 3011 --headless > /workspace/logs/kohya_ss.log 2>&1 &