Upload 2 files

scripts/Model_downloader/Download-HIDREAM-Models.ps1

@@ -0,0 +1,85 @@
+param(
+    [string]$InstallPath = $PSScriptRoot
+)
+
+<#
+.SYNOPSIS
+    A PowerShell script to interactively download HiDream models for ComfyUI.
+.DESCRIPTION
+    This version corrects the download logic for common files (VAE, CLIPs).
+#>
+
+#===========================================================================
+# SECTION 1: HELPER FUNCTIONS & SETUP
+#===========================================================================
+function Write-Log { param([string]$Message, [string]$Color = "White") { $logFile = Join-Path $InstallPath "logs\install_log.txt"; $formattedMessage = "[$([DateTime]::Now.ToString('yyyy-MM-dd HH:mm:ss'))] [ModelDownloader-HiDream] $Message"; Write-Host $Message -ForegroundColor $Color; Add-Content -Path $logFile -Value $formattedMessage -ErrorAction SilentlyContinue; } }
+function Invoke-AndLog { param([string]$File, [string]$Arguments) { $logFile = Join-Path $InstallPath "logs\install_log.txt"; $commandToRun = "`"$File`" $Arguments"; $cmdArguments = "/C `"$commandToRun >> `"`"$logFile`"`" 2>&1`""; try { Start-Process -FilePath "cmd.exe" -ArgumentList $cmdArguments -Wait -WindowStyle Hidden } catch { Write-Log "FATAL ERROR trying to execute command: $commandToRun" -Color Red } } }
+function Download-File { param([string]$Uri, [string]$OutFile) { if (Test-Path $OutFile) { Write-Log "Skipping: $((Split-Path $OutFile -Leaf)) (already exists)." -Color Gray } else { $fileName = Split-Path -Path $Uri -Leaf; if (Get-Command 'aria2c' -ErrorAction SilentlyContinue) { Write-Log "Downloading: $fileName"; $aria_args = "-c -x 16 -s 16 -k 1M --dir=`"$((Split-Path $OutFile -Parent))`" --out=`"$((Split-Path $OutFile -Leaf))`" `"$Uri`""; Invoke-AndLog "aria2c" $aria_args } else { Write-Log "Aria2 not found. Falling back to standard download: $fileName" -Color Yellow; Invoke-WebRequest -Uri $Uri -OutFile $OutFile } } } }
+function Ask-Question { param([string]$Prompt, [string[]]$Choices, [string[]]$ValidAnswers) { $choice = ''; while ($choice -notin $ValidAnswers) { Write-Log "`n$Prompt" -Color Yellow; foreach ($line in $Choices) { Write-Host "  $line" -ForegroundColor Green }; $choice = (Read-Host "Enter your choice and press Enter").ToUpper(); if ($choice -notin $ValidAnswers) { Write-Log "Invalid choice. Please try again." -Color Red } }; return $choice } }
+
+#===========================================================================
+# SECTION 2: SCRIPT EXECUTION
+#===========================================================================
+
+$InstallPath = $InstallPath.Trim('"')
+$comfyPath = Join-Path $InstallPath "ComfyUI"
+$modelsPath = Join-Path $comfyPath "models"
+if (-not (Test-Path $modelsPath)) { Write-Log "Could not find ComfyUI models path at '$modelsPath'. Exiting." -Color Red; Read-Host "Press Enter to exit."; exit }
+
+# --- GPU Detection ---
+Write-Log "-------------------------------------------------------------------------------"
+Write-Log "Checking for NVIDIA GPU to provide model recommendations..." -Color Yellow
+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 24) { Write-Log "Recommandation: fp8" -Color Cyan } elseif ($gpuMemoryGiB -ge 16) { Write-Log "Recommandation: GGUF Q8_0" -Color Cyan } elseif ($gpuMemoryGiB -ge 12) { Write-Log "Recommandation: GGUF Q5_K_S" -Color Cyan } else { Write-Log "Recommandation: GGUF Q4_K_S" -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 ---
+$fp8Choice = Ask-Question "Do you want to download HiDream fp8 models? (24GB Vram recommended)" @("A) Yes", "B) No") @("A", "B")
+$ggufChoice = Ask-Question "Do you want to download HiDream GGUF models?" @("A) Q8_0 (16GB Vram)", "B) Q5_K_S (12GB Vram)", "C) Q4_K_S (less than 12GB Vram)", "D) All", "E) No") @("A", "B", "C", "D", "E")
+
+# --- Download files based on answers ---
+Write-Log "`nStarting HiDream model downloads..." -Color Cyan
+$baseUrl = "https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models"
+$hidreamDiffDir = Join-Path $modelsPath "diffusion_models\HIDREAM"
+$hidreamUnetDir = Join-Path $modelsPath "unet\HIDREAM"
+$clipDir = Join-Path $modelsPath "clip"
+$vaeDir  = Join-Path $modelsPath "vae"
+New-Item -Path $hidreamDiffDir, $hidreamUnetDir, $clipDir, $vaeDir -ItemType Directory -Force | Out-Null
+
+# === CORRECTION : Déterminer si un téléchargement est nécessaire ===
+$doDownload = ($fp8Choice -eq 'A' -or $ggufChoice -ne 'E')
+
+# Télécharger les fichiers communs si nécessaire
+if ($doDownload) {
+    Write-Log "`nDownloading HiDream common support files (VAE, CLIPs)..."
+    Download-File -Uri "$baseUrl/vae/ae.safetensors?download=true" -OutFile (Join-Path $vaeDir "ae.safetensors")
+    Download-File -Uri "$baseUrl/clip/clip_g_hidream.safetensors?download=true" -OutFile (Join-Path $clipDir "clip_g_hidream.safetensors")
+    Download-File -Uri "$baseUrl/clip/clip_l_hidream.safetensors?download=true" -OutFile (Join-Path $clipDir "clip_l_hidream.safetensors")
+    Download-File -Uri "$baseUrl/clip/t5xxl_fp8_e4m3fn_scaled.safetensors?download=true" -OutFile (Join-Path $clipDir "t5xxl_fp8_e4m3fn_scaled.safetensors")
+    Download-File -Uri "$baseUrl/clip/llama_3.1_8b_instruct_fp8_scaled.safetensors?download=true" -OutFile (Join-Path $clipDir "llama_3.1_8b_instruct_fp8_scaled.safetensors")
+}
+
+# fp8 Model (uniquement le modèle principal)
+if ($fp8Choice -eq 'A') {
+    Write-Log "`nDownloading HiDream fp8 model..."
+    Download-File -Uri "$baseUrl/diffusion_models/HiDream/hidream_i1_dev_fp8.safetensors?download=true" -OutFile (Join-Path $hidreamDiffDir "hidream_i1_dev_fp8.safetensors")
+}
+
+# GGUF Models
+if ($ggufChoice -ne 'E') {
+    Write-Log "`nDownloading HiDream GGUF models..."
+    if ($ggufChoice -in 'A', 'D') { Download-File -Uri "$baseUrl/unet/HiDream/hidream-i1-dev-Q8_0.gguf?download=true" -OutFile (Join-Path $hidreamUnetDir "hidream-i1-dev-Q8_0.gguf") }
+    if ($ggufChoice -in 'B', 'D') { Download-File -Uri "$baseUrl/unet/HiDream/hidream-i1-dev-Q5_K_S.gguf?download=true" -OutFile (Join-Path $hidreamUnetDir "hidream-i1-dev-Q5_K_S.gguf") }
+    if ($ggufChoice -in 'C', 'D') { Download-File -Uri "$baseUrl/unet/HiDream/hidream-i1-dev-Q4_K_S.gguf?download=true" -OutFile (Join-Path $hidreamUnetDir "hidream-i1-dev-Q4_K_S.gguf") }
+}
+
+Write-Log "`nHiDream model downloads complete." -Color Green
+Read-Host "Press Enter to return to the main installer."

scripts/Model_downloader/Download-LTXV-Models.ps1

@@ -0,0 +1,85 @@
+param(
+    [string]$InstallPath = $PSScriptRoot
+)
+
+<#
+.SYNOPSIS
+    A PowerShell script to interactively download LTX-Video models for ComfyUI.
+.DESCRIPTION
+    This version corrects the download logic for the common VAE file.
+#>
+
+#===========================================================================
+# SECTION 1: HELPER FUNCTIONS & SETUP
+#===========================================================================
+function Write-Log { param([string]$Message, [string]$Color = "White") { $logFile = Join-Path $InstallPath "logs\install_log.txt"; $formattedMessage = "[$([DateTime]::Now.ToString('yyyy-MM-dd HH:mm:ss'))] [ModelDownloader-LTXV] $Message"; Write-Host $Message -ForegroundColor $Color; Add-Content -Path $logFile -Value $formattedMessage -ErrorAction SilentlyContinue; } }
+function Invoke-AndLog { param([string]$File, [string]$Arguments) { $logFile = Join-Path $InstallPath "logs\install_log.txt"; $commandToRun = "`"$File`" $Arguments"; $cmdArguments = "/C `"$commandToRun >> `"`"$logFile`"`" 2>&1`""; try { Start-Process -FilePath "cmd.exe" -ArgumentList $cmdArguments -Wait -WindowStyle Hidden } catch { Write-Log "FATAL ERROR trying to execute command: $commandToRun" -Color Red } } }
+function Download-File { param([string]$Uri, [string]$OutFile) { if (Test-Path $OutFile) { Write-Log "Skipping: $((Split-Path $OutFile -Leaf)) (already exists)." -Color Gray } else { $fileName = Split-Path -Path $Uri -Leaf; if (Get-Command 'aria2c' -ErrorAction SilentlyContinue) { Write-Log "Downloading: $fileName"; $aria_args = "-c -x 16 -s 16 -k 1M --dir=`"$((Split-Path $OutFile -Parent))`" --out=`"$((Split-Path $OutFile -Leaf))`" `"$Uri`""; Invoke-AndLog "aria2c" $aria_args } else { Write-Log "Aria2 not found. Falling back to standard download: $fileName" -Color Yellow; Invoke-WebRequest -Uri $Uri -OutFile $OutFile } } } }
+function Ask-Question { param([string]$Prompt, [string[]]$Choices, [string[]]$ValidAnswers) { $choice = ''; while ($choice -notin $ValidAnswers) { Write-Log "`n$Prompt" -Color Yellow; foreach ($line in $Choices) { Write-Host "  $line" -ForegroundColor Green }; $choice = (Read-Host "Enter your choice and press Enter").ToUpper(); if ($choice -notin $ValidAnswers) { Write-Log "Invalid choice. Please try again." -Color Red } }; return $choice } }
+
+#===========================================================================
+# SECTION 2: SCRIPT EXECUTION
+#===========================================================================
+
+$InstallPath = $InstallPath.Trim('"')
+$comfyPath = Join-Path $InstallPath "ComfyUI"
+$modelsPath = Join-Path $comfyPath "models"
+if (-not (Test-Path $modelsPath)) { Write-Log "Could not find ComfyUI models path at '$modelsPath'. Exiting." -Color Red; Read-Host "Press Enter to exit."; exit }
+
+# --- GPU Detection ---
+Write-Log "-------------------------------------------------------------------------------"
+Write-Log "Checking for NVIDIA GPU to provide model recommendations..." -Color Yellow
+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: 13B Base Model" -Color Cyan } elseif ($gpuMemoryGiB -ge 24) { Write-Log "Recommandation: GGUF Q8_0" -Color Cyan } elseif ($gpuMemoryGiB -ge 16) { Write-Log "Recommandation: GGUF Q5_K_M" -Color Cyan } else { Write-Log "Recommandation: 2B Base Model ou GGUF Q3_K_S" -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 ---
+$baseChoice = Ask-Question "Do you want to download LTXV base models?" @("A) 13B (30Gb)", "B) 2B (7Gb)", "C) All", "D) No") @("A", "B", "C", "D")
+$ggufChoice = Ask-Question "Do you want to download LTXV GGUF models?" @("A) Q8_0 (24GB Vram)", "B) Q5_K_M (16GB Vram)", "C) Q3_K_S (less than 12GB Vram)", "D) All", "E) No") @("A", "B", "C", "D", "E")
+
+# --- Download files based on answers ---
+Write-Log "`nStarting LTX-Video model downloads..." -Color Cyan
+$baseUrl = "https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models"
+$ltxvChkptDir = Join-Path $modelsPath "checkpoints\LTXV"
+$ltxvUnetDir  = Join-Path $modelsPath "unet\LTXV"
+$vaeDir       = Join-Path $modelsPath "vae"
+New-Item -Path $ltxvChkptDir, $ltxvUnetDir, $vaeDir -ItemType Directory -Force | Out-Null
+
+# === CORRECTION : Déterminer si un téléchargement est nécessaire pour les fichiers communs ===
+$doDownload = ($baseChoice -ne 'D' -or $ggufChoice -ne 'E')
+
+# Télécharger le VAE commun si N'IMPORTE QUEL modèle est sélectionné
+if ($doDownload) {
+    Write-Log "`nDownloading LTXV common support file (VAE)..."
+    Download-File -Uri "$baseUrl/vae/ltxv-13b-0.9.7-vae-BF16.safetensors?download=true" -OutFile (Join-Path $vaeDir "ltxv-13b-0.9.7-vae-BF16.safetensors")
+}
+
+# Base Models (uniquement les checkpoints)
+if ($baseChoice -ne 'D') {
+    Write-Log "`nDownloading LTXV base model(s)..."
+    if ($baseChoice -in 'A', 'C') {
+        Download-File -Uri "$baseUrl/checkpoints/LTXV/ltxv-13b-0.9.7-dev.safetensors?download=true" -OutFile (Join-Path $ltxvChkptDir "ltxv-13b-0.9.7-dev.safetensors")
+    }
+    if ($baseChoice -in 'B', 'C') {
+        Download-File -Uri "$baseUrl/checkpoints/LTXV/ltxv-2b-0.9.6-dev-04-25.safetensors?download=true" -OutFile (Join-Path $ltxvChkptDir "ltxv-2b-0.9.6-dev-04-25.safetensors")
+    }
+}
+
+# GGUF Models
+if ($ggufChoice -ne 'E') {
+    Write-Log "`nDownloading LTXV GGUF models..."
+    if ($ggufChoice -in 'A', 'D') { Download-File -Uri "$baseUrl/unet/LTXV/ltxv-13b-0.9.7-dev-Q8_0.gguf?download=true" -OutFile (Join-Path $ltxvUnetDir "ltxv-13b-0.9.7-dev-Q8_0.gguf") }
+    if ($ggufChoice -in 'B', 'D') { Download-File -Uri "$baseUrl/unet/LTXV/ltxv-13b-0.9.7-dev-Q5_K_M.gguf?download=true" -OutFile (Join-Path $ltxvUnetDir "ltxv-13b-0.9.7-dev-Q5_K_M.gguf") }
+    if ($ggufChoice -in 'C', 'D') { Download-File -Uri "$baseUrl/unet/LTXV/ltxv-13b-0.9.7-dev-Q3_K_S.gguf?download=true" -OutFile (Join-Path $ltxvUnetDir "ltxv-13b-0.9.7-dev-Q3_K_S.gguf") }
+}
+
+Write-Log "`nLTX-Video model downloads complete." -Color Green
+Read-Host "Press Enter to return to the main installer."