Update build.sh

build.sh

@@ -1,91 +1,26 @@
 #!/bin/bash
 set -e
 
-echo "=========================================="
-echo "🏗️  BUILD PHASE: Installing Dependencies"
-echo "=========================================="
-echo ""
+echo "📦 Installing ComfyUI + GGUF support..."
 
-# Step 1: System dependencies
-echo "[1/5] Installing system packages..."
-apt-get update > /dev/null 2>&1
-apt-get install -y git wget curl build-essential cmake > /dev/null 2>&1
-echo "✓ System deps installed"
+# Install ComfyUI dependencies
+pip install -r ComfyUI/requirements.txt -q
 
-# Step 2: Clone ComfyUI
-echo ""
-echo "[2/5] Cloning ComfyUI..."
-if [ ! -d "ComfyUI" ]; then
-    git clone --depth 1 https://github.com/comfyanonymous/ComfyUI.git
-fi
-cd ComfyUI
-pip install -r requirements.txt -q
-cd ..
-echo "✓ ComfyUI installed"
+# Install GGUF node
+git clone --depth 1 https://github.com/city96/ComfyUI-GGUF.git ComfyUI/custom_nodes/ComfyUI-GGUF
+pip install -r ComfyUI/custom_nodes/ComfyUI-GGUF/requirements.txt -q
 
-# Step 3: Install ComfyUI-GGUF Node
-echo ""
-echo "[3/5] Installing ComfyUI-GGUF node..."
-mkdir -p ComfyUI/custom_nodes
-if [ ! -d "ComfyUI/custom_nodes/ComfyUI-GGUF" ]; then
-    git clone --depth 1 https://github.com/city96/ComfyUI-GGUF.git ComfyUI/custom_nodes/ComfyUI-GGUF
-fi
-if [ -f "ComfyUI/custom_nodes/ComfyUI-GGUF/requirements.txt" ]; then
-    pip install -r ComfyUI/custom_nodes/ComfyUI-GGUF/requirements.txt -q
-fi
-echo "✓ ComfyUI-GGUF installed"
+# Download model
+mkdir -p models ComfyUI/models/checkpoints
+wget -O models/flux-2-klein-4b-Q4_K_M.gguf \
+  "https://huggingface.co/unsloth/FLUX.2-klein-4B-GGUF/resolve/main/flux-2-klein-4b-Q4_K_M.gguf" \
+  --progress=bar:force:noscroll -q
+cp models/*.gguf ComfyUI/models/checkpoints/
 
-# Step 4: Download Model
-echo ""
-echo "[4/5] Downloading FLUX model..."
-mkdir -p models
-mkdir -p ComfyUI/models/checkpoints
+# Download VAE + CLIP
+wget -O ComfyUI/models/vae/ae.safetensors \
+  "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/ae.safetensors" -q
+wget -O ComfyUI/models/clip/t5xxl_fp16.safetensors \
+  "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp16.safetensors" -q
 
-MODEL_FILE="flux-2-klein-4b-Q4_K_M.gguf"
-URL="https://huggingface.co/unsloth/FLUX.2-klein-4B-GGUF/resolve/main/${MODEL_FILE}"
-
-if [ ! -f "models/${MODEL_FILE}" ]; then
-    echo "   Downloading ${MODEL_FILE}..."
-    wget -O "models/${MODEL_FILE}" "${URL}" --progress=bar:force:noscroll --no-check-certificate
-else
-    echo "   Model already exists"
-fi
-
-# Copy to ComfyUI directory
-cp "models/${MODEL_FILE}" "ComfyUI/models/checkpoints/"
-SIZE=$(du -h "models/${MODEL_FILE}" | cut -f1)
-echo "✓ Model downloaded (${SIZE})"
-
-# Step 5: Download VAE and CLIP (needed for Flux workflow)
-echo ""
-echo "[5/5] Downloading supporting models..."
-
-# VAE
-VAE_URL="https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/ae.safetensors"
-mkdir -p ComfyUI/models/vae
-if [ ! -f "ComfyUI/models/vae/ae.safetensors" ]; then
-    wget -O "ComfyUI/models/vae/ae.safetensors" "$VAE_URL" -q
-fi
-echo "✓ VAE downloaded"
-
-# CLIP Text Encoder (T5)
-CLIP_URL="https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp16.safetensors"
-mkdir -p ComfyUI/models/clip
-if [ ! -f "ComfyUI/models/clip/t5xxl_fp16.safetensors" ]; then
-    wget -O "ComfyUI/models/clip/t5xxl_fp16.safetensors" "$CLIP_URL" -q
-fi
-echo "✓ CLIP encoder downloaded"
-
-echo ""
-echo "=========================================="
-echo "✅ BUILD COMPLETE!"
-echo "=========================================="
-echo ""
-echo "Installed:"
-echo "  • ComfyUI"
-echo "  • ComfyUI-GGUF node"
-echo "  • Flux model (${MODEL_FILE})"
-echo "  • VAE (ae.safetensors)"
-echo "  • CLIP (t5xxl_fp16.safetensors)"
-echo ""
-echo "Ready to generate images!"
+echo "✅ Done!"