raubatz commited on
Commit
f366d80
·
verified ·
1 Parent(s): 6b24fba

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +12 -9
Dockerfile CHANGED
@@ -73,22 +73,25 @@ RUN echo "Copying custom nodes from public bucket..." && \
73
  # Run install scripts
74
  RUN cd /data/ComfyUI/custom_nodes/comfyui_controlnet_aux && pip install -r requirements.txt || true
75
  RUN cd /data/ComfyUI/custom_nodes/stability-ComfyUI-nodes && pip install -r requirements.txt || true
 
76
  # Startup script (run directly as user)
77
  COPY <<'EOF' /start.sh
78
  #!/bin/bash
 
 
 
79
  echo "=== Starting model copy from public bucket ==="
80
  if [ -d "/datapublic/comfy" ]; then
81
- echo "Copying key models for basic SD1.5 workflow..."
82
-
83
- cp -n /datapublic/comfy/stabilityai/sd-vae-ft-mse-original/vae-ft-mse-840000-ema-pruned.safetensors /data/ComfyUI/models/vae/ 2>/dev/null || true
84
- cp -n /datapublic/comfy/LA0077/v1-5-pruned-emaonly-fp16.safetensors/v1-5-pruned-emaonly-fp16.safetensors /data/ComfyUI/models/checkpoints/ 2>/dev/null || true
85
-
86
  echo "Model copy finished."
87
  else
88
- echo "Warning: /datapublic/comfy not found."
89
  fi
90
-
91
- echo "Starting ComfyUI..."
92
  cd /data/ComfyUI
93
- exec python main.py --listen 0.0.0.0 --port 7860 --output-directory "${USE_PERSISTENT_DATA:+/data/}"
94
  EOF
 
 
 
73
  # Run install scripts
74
  RUN cd /data/ComfyUI/custom_nodes/comfyui_controlnet_aux && pip install -r requirements.txt || true
75
  RUN cd /data/ComfyUI/custom_nodes/stability-ComfyUI-nodes && pip install -r requirements.txt || true
76
+
77
  # Startup script (run directly as user)
78
  COPY <<'EOF' /start.sh
79
  #!/bin/bash
80
+ echo "=== Container started as user $(whoami) ==="
81
+ echo "Current directory: $(pwd)"
82
+ ls -la /data/ComfyUI 2>&1 | cat
83
  echo "=== Starting model copy from public bucket ==="
84
  if [ -d "/datapublic/comfy" ]; then
85
+ echo "Bucket found. Copying models..."
86
+ cp -n /datapublic/comfy/stabilityai/sd-vae-ft-mse-original/vae-ft-mse-840000-ema-pruned.safetensors /data/ComfyUI/models/vae/ 2>&1 | cat || true
87
+ cp -n /datapublic/comfy/LA0077/v1-5-pruned-emaonly-fp16.safetensors/v1-5-pruned-emaonly-fp16.safetensors /data/ComfyUI/models/checkpoints/ 2>&1 | cat || true
 
 
88
  echo "Model copy finished."
89
  else
90
+ echo "WARNING: /datapublic/comfy not found!"
91
  fi
92
+ echo "=== Starting ComfyUI server ==="
 
93
  cd /data/ComfyUI
94
+ exec python main.py --listen 0.0.0.0 --port 7860 --output-directory "${USE_PERSISTENT_DATA:+/data/}" 2>&1
95
  EOF
96
+
97
+