From a6f942bd968a19a9502045019daa866ae86b4530 Mon Sep 17 00:00:00 2001 From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com> Date: Sun, 5 Oct 2025 06:20:32 +0100 Subject: [PATCH] . --- install.ps1 | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/install.ps1 b/install.ps1 index 9aaf700f..b5a4413d 100644 --- a/install.ps1 +++ b/install.ps1 @@ -228,7 +228,9 @@ switch ($Device) { Invoke-Pip torch torchaudio --index-url "https://download.pytorch.org/whl/cu128" Invoke-Conda cuda-nvcc=12.8 Invoke-Pip psutil ninja packaging wheel "setuptools>=42" + Write-Info "Installing Flash Attn..." Invoke-Pip flash-attn -i https://xxxxrt666.github.io/PIP-Index/ --no-build-isolation + Write-Success "Flash Attn Installed" } "CU126" { $cudaLine = nvidia-smi | Select-String "CUDA Version" @@ -241,7 +243,9 @@ switch ($Device) { Invoke-Pip torch torchaudio --index-url "https://download.pytorch.org/whl/cu126" Invoke-Conda cuda-nvcc=12.6 Invoke-Pip psutil ninja packaging wheel "setuptools>=42" + Write-Info "Installing Flash Attn..." Invoke-Pip flash-attn -i https://xxxxrt666.github.io/PIP-Index/ --no-build-isolation + Write-Success "Flash Attn Installed" } "CPU" { Write-Info "Installing PyTorch For CPU..."