raubatz commited on
Commit
849b0ad
·
verified ·
1 Parent(s): f366d80

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +13 -12
Dockerfile CHANGED
@@ -75,23 +75,24 @@ RUN cd /data/ComfyUI/custom_nodes/comfyui_controlnet_aux && pip install -r requi
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
-
 
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
+ # Simple startup (run as user)
79
  COPY <<'EOF' /start.sh
80
  #!/bin/bash
81
+ echo "=== Container started as $(whoami) ==="
82
+ echo "Current dir: $(pwd)"
83
+ ls -la /data/ComfyUI 2>&1
84
+
85
+ echo "=== Copying models ==="
86
  if [ -d "/datapublic/comfy" ]; then
87
+ cp -n /datapublic/comfy/stabilityai/sd-vae-ft-mse-original/vae-ft-mse-840000-ema-pruned.safetensors /data/ComfyUI/models/vae/ 2>&1 || true
88
+ cp -n /datapublic/comfy/LA0077/v1-5-pruned-emaonly-fp16.safetensors/v1-5-pruned-emaonly-fp16.safetensors /data/ComfyUI/models/checkpoints/ 2>&1 || true
89
+ echo "Models copied."
 
90
  else
91
+ echo "No bucket found."
92
  fi
93
+
94
+ echo "=== Starting ComfyUI ==="
95
  cd /data/ComfyUI
96
  exec python main.py --listen 0.0.0.0 --port 7860 --output-directory "${USE_PERSISTENT_DATA:+/data/}" 2>&1
97
  EOF
98
+ RUN chmod +x /start.sh