Create build.sh

build.sh

@@ -0,0 +1,91 @@
+#!/bin/bash
+set -e
+
+echo "=========================================="
+echo "🏗️  BUILD PHASE: Installing Dependencies"
+echo "=========================================="
+echo ""
+
+# 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"
+
+# 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"
+
+# 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"
+
+# Step 4: Download Model
+echo ""
+echo "[4/5] Downloading FLUX model..."
+mkdir -p models
+mkdir -p ComfyUI/models/checkpoints
+
+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!"