Upload Install-ComfyUI.ps1

scripts/ComfyUI/Install-ComfyUI.ps1

@@ -264,9 +264,45 @@ foreach ($node in $customNodes) {
 }
 
 # --- Étape 6: Installation des modules Python supplémentaires ---
-Write-Log "`nStep 6: Installing supplementary Python modules..." -Color Yellow
+Write-Log "`nStep 6: Installing supplementary modules..." -Color Yellow
+
+# VS Build Tools
+Write-Log "  - Installing Visual Studio Build Tools (may be required for some packages)..."
+$wingetArgs = "install --id Microsoft.VisualStudio.2022.BuildTools -e --source winget --override --quiet --wait --norestart --add Microsoft.VisualStudio.Component.VC.Tools.x86.x64 --add Microsoft.VisualStudio.Component.Windows10SDK.20348"
+Invoke-AndLog "winget" $wingetArgs
+
+# Python include/libs
+Write-Log "  - Installing Python include/libs for compatibility..."
+$pyLibsUrl = "https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/others/python_3.12.9_include_libs.zip"
+$zipPath = Join-Path $InstallPath "python_libs.tmp.zip"
+$extractPath = Join-Path $comfyPath "venv" # On extrait dans le venv
+Download-File -Uri $pyLibsUrl -OutFile $zipPath
+if (Test-Path $zipPath) {
+    try {
+        Expand-Archive -Path $zipPath -DestinationPath $extractPath -Force
+        Write-Log "    - Python libs extracted successfully." -Color Green
+    } catch { Write-Log "    - FAILED to extract python libs. Error: $($_.Exception.Message)" -Color Red } finally { Remove-Item -Path $zipPath -Force }
+}
+
+# Triton
+Write-Log "  - Installing Triton..."
+$tritonWheel = Join-Path $InstallPath "triton.whl"
+Download-File -Uri "https://github.com/woct0rdho/triton-windows/releases/download/empty/triton-3.3.0-py3-none-any.whl" -OutFile $tritonWheel
+Invoke-AndLog "$venvPython" "-m pip install `"$tritonWheel`""
+Invoke-AndLog "$venvPython" "-m pip install triton-windows"
+Remove-Item $tritonWheel -ErrorAction SilentlyContinue
+
+# xformers
+Write-Log "  - Installing xformers..."
 Invoke-AndLog "$venvPython" "-m pip install -U xformers --index-url https://download.pytorch.org/whl/cu128"
-# ... (Ajoutez ici l'installation de Triton, SageAttention etc. si nécessaire, en utilisant $venvPython) ...
+# La logique de renommage xformers n'est généralement plus nécessaire avec les installations pip modernes, mais peut être rajoutée ici si besoin.
+
+# SageAttention
+Write-Log "  - Installing SageAttention..."
+$sageWheel = Join-Path $InstallPath "sageattention.whl"
+Download-File -Uri "https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/whl/sageattention-2.1.1+cu128torch2.7.0-cp312-cp312-win_amd64.whl?download=true" -OutFile $sageWheel
+Invoke-AndLog "$venvPython" "-m pip install `"$sageWheel`""
+Remove-Item $sageWheel -ErrorAction SilentlyContinue
 
 # --- Étape 7: Téléchargement des Workflows et Settings ---
 Write-Log "`nStep 7: Downloading Workflows & Settings..." -Color Yellow