.

Author: XXXXRT666

.github/build_windows_packages.ps1

@@ -117,9 +117,9 @@ Write-Host "[INFO] Installing PyTorch..."
 & ".\runtime\python.exe" -m pip install --upgrade pip --no-warn-script-location
 
 switch ($cuda) {
-    "cu124" {
+    "cu126" {
         & ".\runtime\python.exe" -m pip install psutil ninja packaging wheel "setuptools>=42" --no-warn-script-location
-        & ".\runtime\python.exe" -m pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu124 --no-warn-script-location
+        & ".\runtime\python.exe" -m pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu126 --no-warn-script-location
         & ".\runtime\python.exe" -m pip install flash-attn -i https://xxxxrt666.github.io/PIP-Index/ --no-build-isolation
     }
     "cu128" {

.github/workflows/build_windows_packages.yaml

@@ -17,7 +17,7 @@ jobs:
     runs-on: windows-latest
     strategy:
       matrix:
-        torch_cuda: [cu124, cu128]
+        torch_cuda: [cu126, cu128]
     env:
       TORCH_CUDA: ${{ matrix.torch_cuda }}
       MODELSCOPE_USERNAME: ${{ secrets.MODELSCOPE_USERNAME }}
@@ -32,7 +32,6 @@ jobs:
         uses: actions/checkout@v4
 
       - name: Install Windows CUDA 12.9
-        if: ${{ runner.os == 'Windows' && matrix.torch_cuda == '12.8' }}
         uses: Jimver/cuda-toolkit
         id: cuda-toolkit-win-129
         with: