Update script.sh

script.sh

@@ -55,8 +55,19 @@ download_file() {
 
         if command -v aria2c >/dev/null 2>&1; then
             log_info "使用 aria2c (3 個連線) 進行下載..."
-            # 使用 -x 3 -s 3，並允許覆蓋，不重複檢查
-            aria2c --console-log-level=error -c -x 3 -s 3 -k 1M --check-certificate=false --allow-overwrite=true -d "$(dirname "$dest_path")" -o "$filename" "$url"
+            aria2c --console-log-level=error \
+                -c -x 16 -s 16 -k 1M \
+                --max-connection-per-server=16 \
+                --min-split-size=1M \
+                --max-tries=5 \
+                --retry-wait=3 \
+                --timeout=60 \
+                --check-certificate=false \
+                --allow-overwrite=true \
+                --file-allocation=none \
+                -d "$(dirname "$dest_path")" \
+                -o "$filename" \
+                "$url"
         else
             log_info "使用 wget 進行下載..."
             wget -c -O "$dest_path" --quiet --show-progress "$url"
@@ -76,13 +87,57 @@ download_file() {
     return 0
 }
 
+download_models_pack() {
+    log_info "[並行任務] 開始下載模型包..."
+    
+    local MODELS_ZIP_URL="https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/backup_models/models_pack.zip"
+    local ZIP_PATH="/workspace/ComfyUI/models_pack.zip"
+    local MAX_ATTEMPTS=2
+    local ATTEMPT=1
+    local SUCCESS=false
+
+    while [ $ATTEMPT -le $MAX_ATTEMPTS ] && [ "$SUCCESS" = false ]; do
+        log_info "[模型包] 嘗試 ${ATTEMPT}/${MAX_ATTEMPTS}..."
+
+        download_file "$ZIP_PATH" "$MODELS_ZIP_URL"
+        
+        if [ $? -eq 0 ]; then
+            log_info "[模型包] 正在解壓縮..."
+            unzip -o -q "$ZIP_PATH" -d /workspace/ComfyUI/
+            
+            if [ $? -eq 0 ]; then
+                log_success "[模型包] 成功解壓縮。"
+                SUCCESS=true
+                rm -f "$ZIP_PATH"
+                log_info "[模型包] 已刪除壓縮檔。"
+            else
+                log_error "[模型包] 解壓縮失敗！"
+                rm -f "$ZIP_PATH"
+                ATTEMPT=$((ATTEMPT + 1))
+                [ $ATTEMPT -le $MAX_ATTEMPTS ] && sleep 5
+            fi
+        else
+            ATTEMPT=$((ATTEMPT + 1))
+        fi
+    done
+
+    if [ "$SUCCESS" = false ]; then
+        log_error "[模型包] 下載失敗，已達最大重試次數。"
+        echo "MODELS_DOWNLOAD_FAILED" > /tmp/download_status
+        return 1
+    fi
+    
+    echo "MODELS_DOWNLOAD_SUCCESS" > /tmp/download_status
+    return 0
+}
+
 log_info "Provisioning Started at $(date)"
 
-log_step "1/5" "運行 Vast.ai 基礎腳本"
+log_step "1/4" "運行 Vast.ai 基礎腳本"
 wget -q -O /tmp/default.sh https://raw.githubusercontent.com/vast-ai/base-image/refs/heads/main/derivatives/pytorch/derivatives/comfyui/provisioning_scripts/default.sh
 bash /tmp/default.sh || log_warn "基礎腳本遇到非致命錯誤，繼續執行..."
 
-log_step "2/5" "準備 ComfyUI 目錄與環境"
+log_step "2/4" "準備 ComfyUI 目錄與環境"
 cd /workspace/ComfyUI || { log_error "找不到 ComfyUI 目錄，中止執行。"; exit 1; }
 
 log_info "檢測 Python 環境..."
@@ -96,7 +151,15 @@ else
     log_warn "使用系統 Python: $PYTHON_BIN"
 fi
 
-log_step "3/5" "安裝 ComfyUI 自定義節點與依賴"
+log_step "3/4" "並行處理：節點安裝 + 模型下載"
+
+# ============ 啟動並行下載任務 ============
+log_info "🚀 啟動模型包下載（後台任務）..."
+download_models_pack &
+DOWNLOAD_PID=$!
+
+# ============ 同時進行節點安裝 ============
+log_info "🚀 開始安裝自定義節點（前台任務）..."
 cd custom_nodes
 
 install_node "https://github.com/ltdrdata/ComfyUI-Impact-Pack.git"
@@ -134,51 +197,33 @@ done
 
 cd /workspace/ComfyUI
 
-log_step "4/5" "下載並解壓縮模型包"
-MODELS_ZIP_URL="https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/backup_models/models_pack.zip"
-ZIP_PATH="/workspace/ComfyUI/models_pack.zip"
+log_info "⏳ 等待模型包下載完成..."
+wait $DOWNLOAD_PID
+DOWNLOAD_EXIT_CODE=$?
 
-download_file "user/default/workflows/Single_Posture.json" "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/Single_Posture.json?download=true"
+if [ -f /tmp/download_status ]; then
+    DOWNLOAD_STATUS=$(cat /tmp/download_status)
+    if [ "$DOWNLOAD_STATUS" = "MODELS_DOWNLOAD_FAILED" ]; then
+        log_error "模型包下載失敗，請檢查網路或稍後重試。"
+        exit 1
+    fi
+fi
 
-MAX_UNZIP_ATTEMPTS=2
-UNZIP_ATTEMPT=1
-UNZIP_SUCCESS=false
+log_success "所有並行任務已完成！"
 
-while [ $UNZIP_ATTEMPT -le $MAX_UNZIP_ATTEMPTS ] && [ "$UNZIP_SUCCESS" = false ]; do
-    log_info "--- [第 ${UNZIP_ATTEMPT}/${MAX_UNZIP_ATTEMPTS} 次嘗試] ---"
+download_file "user/default/workflows/Single_Posture.json" "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/Single_Posture.json?download=true"
 
-    # 步驟 1: 下載壓縮檔
-    download_file "$ZIP_PATH" "$MODELS_ZIP_URL"
-    if [ $? -ne 0 ]; then
-        log_error "模型壓縮檔下載失敗，無法繼續。"
-        exit 1
-    fi
+log_step "4/4" "清理與最終配置"
 
-    # 步驟 2: 解壓縮
-    log_info "正在將 'models_pack.zip' 解壓縮至 ComfyUI 目錄並覆蓋現有檔案..."
-    # 使用 -o 選項來無條件覆蓋檔案
-    unzip -o "$ZIP_PATH" -d /workspace/ComfyUI/
-    
-    if [ $? -eq 0 ]; then
-        log_success "成功解壓縮模型包。"
-        UNZIP_SUCCESS=true
-        # 成功後刪除壓縮檔以釋放空間
-        rm -f "$ZIP_PATH"
-        log_info "已刪除 'models_pack.zip'。"
-    else
-        log_error "解壓縮失敗！壓縮檔未下載完成。"
-        UNZIP_ATTEMPT=$((UNZIP_ATTEMPT + 1))
-        if [ $UNZIP_ATTEMPT -le $MAX_UNZIP_ATTEMPTS ]; then
-            log_warn "正在刪除損壞的壓縮檔並準備重新下載..."
-            rm -f "$ZIP_PATH"
-            sleep 5
-        else
-            log_error "已達最大重試次數，解壓縮失敗，可能是因為該檔案被下載的太密集。請取消伺服器租賃、稍等後再重新啟動流程。"
-            exit 1
-        fi
-    fi
-done
+if [ -f "/workspace/ComfyUI/input/example.jpg" ]; then
+    log_info "正在刪除 example.jpg..."
+    rm -f /workspace/ComfyUI/input/example.jpg
+    log_success "已刪除 example.jpg。"
+else
+    log_info "example.jpg 不存在，無需刪除。"
+fi
 
-log_step "5/5" "環境配置完成"
+log_success "環境配置完成！"
 log_info "ComfyUI 即將啟動。實例已準備就緒。"
-echo ""
+echo ""
+echo "📊 總耗時: $(($(date +%s) - $(date -d "$(head -1 /tmp/provision_start_time 2>/dev/null || echo '0 seconds ago')" +%s 2>/dev/null || echo 0))) 秒"