Upload Download-FLUX-Models.ps1

scripts/Model_downloader/Download-FLUX-Models.ps1

@@ -8,7 +8,8 @@ param(
 .SYNOPSIS
     A fully refactored and corrected PowerShell script to interactively download FLUX models for ComfyUI.
 .DESCRIPTION
-    This version corrects pathing issues, download logic, and optimizes directory creation.
+    This version corrects logging paths when called from a parent script, fixes all download logic,
+    and provides user guidance based on GPU VRAM.
 #>
 
 #===========================================================================
@@ -16,15 +17,17 @@ param(
 #===========================================================================
 function Write-Log { 
     param([string]$Message, [string]$Color = "White") 
-    $logFile = Join-Path (Split-Path $MyInvocation.MyCommand.Path -Parent) "logs\install_log.txt"
+    # Utilise $InstallPath (passé en argument) pour trouver le bon dossier de logs
+    $logFile = Join-Path $InstallPath "logs\install_log.txt"
     $formattedMessage = "[$([DateTime]::Now.ToString('yyyy-MM-dd HH:mm:ss'))] [ModelDownloader] $Message"
     Write-Host $Message -ForegroundColor $Color
-    Add-Content -Path $logFile -Value $formattedMessage
+    Add-Content -Path $logFile -Value $formattedMessage -ErrorAction SilentlyContinue
 }
 
 function Invoke-AndLog { 
     param([string]$File, [string]$Arguments) 
-    $logFile = Join-Path (Split-Path $MyInvocation.MyCommand.Path -Parent) "logs\install_log.txt"
+    # Utilise $InstallPath pour trouver le bon dossier de logs
+    $logFile = Join-Path $InstallPath "logs\install_log.txt"
     $commandToRun = "`"$File`" $Arguments"
     $cmdArguments = "/C `"$commandToRun >> `"`"$logFile`"`" 2>&1`""
     try { 
@@ -67,7 +70,6 @@ function Ask-Question {
     return $choice 
 }
 
-
 #===========================================================================
 # SECTION 2: SCRIPT EXECUTION
 #===========================================================================
@@ -88,7 +90,40 @@ if (-not (Test-Path $modelsPath)) {
 # --- GPU Detection ---
 Write-Log "-------------------------------------------------------------------------------"
 Write-Log "Checking for NVIDIA GPU to provide model recommendations..." -Color Yellow
-# ... (logique de détection GPU inchangée) ...
+if (Get-Command 'nvidia-smi' -ErrorAction SilentlyContinue) {
+    try {
+        $gpuInfoCsv = nvidia-smi --query-gpu=name,memory.total --format=csv,noheader
+        if ($gpuInfoCsv) {
+            $gpuInfoParts = $gpuInfoCsv.Split(',')
+            $gpuName = $gpuInfoParts[0].Trim()
+            $gpuMemoryMiB = ($gpuInfoParts[1] -replace ' MiB').Trim()
+            $gpuMemoryGiB = [math]::Round([int]$gpuMemoryMiB / 1024)
+            
+            Write-Log "GPU : $gpuName" -Color Green
+            Write-Log "VRAM : $gpuMemoryGiB GB" -Color Green
+            
+            if ($gpuMemoryGiB -ge 30) {
+                Write-Log "Recommandation: fp16" -Color Cyan
+            } elseif ($gpuMemoryGiB -ge 18) {
+                Write-Log "Recommandation: fp8 or GGUF Q8" -Color Cyan
+            } elseif ($gpuMemoryGiB -ge 14) {
+                Write-Log "Recommandation: GGUF Q6" -Color Cyan
+            }elseif ($gpuMemoryGiB -ge 12) {
+                Write-Log "Recommandation: GGUF Q5" -Color Cyan
+            }elseif ($gpuMemoryGiB -ge 10) {
+                Write-Log "Recommandation: GGUF Q4" -Color Cyan
+            }elseif ($gpuMemoryGiB -ge 8) {
+                Write-Log "Recommandation: GGUF Q3" -Color Cyan
+            }else {
+                Write-Log "Recommandation: GGUF Q2" -Color Cyan
+            }
+        }
+    } catch {
+        Write-Log "Impossible de récupérer les informations GPU. Erreur: $($_.Exception.Message)" -Color Red
+    }
+} else {
+    Write-Log "Aucun GPU NVIDIA detecte (nvidia-smi introuvable). Choisissez selon votre matériel." -Color Gray
+}
 Write-Log "-------------------------------------------------------------------------------"
 
 # --- Ask all questions first ---
@@ -99,13 +134,10 @@ $controlnetChoice = Ask-Question -Prompt "Do you want to download FLUX ControlNe
 $pulidChoice = Ask-Question -Prompt "Do you want to download FLUX PuLID and REDUX Models?" -Choices @("A) Yes", "B) No") -ValidAnswers @("A", "B")
 $loraChoice = Ask-Question -Prompt "Do you want to download UmeAiRT LoRAs?" -Choices @("A) Yes", "B) No") -ValidAnswers @("A", "B")
 
-
-# ===========================================================================
-# ===                     SECTION DE TÉLÉCHARGEMENT CORRIGÉE                ===
-# ===========================================================================
+# --- Download files based on answers ---
 Write-Log "`nStarting downloads based on your choices..." -Color Cyan
 
-# --- 1. Définir tous les chemins une seule fois ---
+# Définir tous les chemins une seule fois
 $baseUrl = "https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models"
 $fluxDir = Join-Path $modelsPath "diffusion_models\FLUX"
 $clipDir = Join-Path $modelsPath "clip"
@@ -116,7 +148,7 @@ $pulidDir = Join-Path $modelsPath "pulid"
 $styleDir = Join-Path $modelsPath "style_models"
 $loraDir = Join-Path $modelsPath "loras\FLUX"
 
-# --- 2. Créer tous les dossiers nécessaires en une seule fois ---
+# Créer tous les dossiers nécessaires en une seule fois
 $requiredDirs = @($fluxDir, $clipDir, $vaeDir, $unetFluxDir, $controlnetDir, $pulidDir, $styleDir, $loraDir)
 foreach ($dir in $requiredDirs) {
     if (-not (Test-Path $dir)) {
@@ -124,10 +156,11 @@ foreach ($dir in $requiredDirs) {
     }
 }
 
-# --- 3. Exécuter les téléchargements avec la logique corrigée ---
+# Vérifier si un téléchargement est nécessaire pour télécharger les fichiers communs
+$doDownload = ($fluxChoice -ne 'D' -or $ggufChoice -ne 'H' -or $schnellChoice -eq 'A' -or $controlnetChoice -ne 'G' -or $pulidChoice -eq 'A' -or $loraChoice -eq 'A')
 
-# Fichiers communs (si un téléchargement est demandé)
-if ($fluxChoice -ne 'D' -or $ggufChoice -ne 'H' -or $schnellChoice -eq 'A' -or $controlnetChoice -ne 'G' -or $pulidChoice -eq 'A' -or $loraChoice -eq 'A') {
+if ($doDownload) {
+    Write-Log "`nDownloading common support models (VAE, CLIP)..."
     Download-File -Uri "$baseUrl/vae/ae.safetensors" -OutFile (Join-Path $vaeDir "ae.safetensors")
     Download-File -Uri "$baseUrl/clip/clip_l.safetensors" -OutFile (Join-Path $clipDir "clip_l.safetensors")
 }
@@ -157,7 +190,7 @@ if ($schnellChoice -eq 'A') {
 
 # ControlNet Models
 if ($controlnetChoice -ne 'G') {
-    if ($controlnetChoice -in 'A','B','F') { # fp16 et fp8 partagent ces controlnets
+    if ($controlnetChoice -in 'A','B','F') { 
         Download-File -Uri "$baseUrl/xlabs/controlnets/flux-canny-controlnet-v3.safetensors" -OutFile (Join-Path $controlnetDir "flux-canny-controlnet-v3.safetensors")
         Download-File -Uri "$baseUrl/xlabs/controlnets/flux-depth-controlnet-v3.safetensors" -OutFile (Join-Path $controlnetDir "flux-depth-controlnet-v3.safetensors")
     }