From 03f8aeec736065983e6a6f6709c089bfacaab625 Mon Sep 17 00:00:00 2001
From: jackal <29866357+ssteo@users.noreply.github.com>
Date: Tue, 15 Jul 2025 13:05:55 +0200
Subject: [PATCH] torch and sageattention2 version need to match

This is needed to start wan2gp, otherwise it will fail with "Torch not being compiled with CUDA enabled"
---
 torch.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/torch.js b/torch.js
index b55fe15..54315de 100644
--- a/torch.js
+++ b/torch.js
@@ -8,9 +8,9 @@ module.exports = {
         "venv": "{{args && args.venv ? args.venv : null}}",
         "path": "{{args && args.path ? args.path : '.'}}",
         "message": [
-          'uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128',
+          'uv pip install torch==2.7.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128',
           "uv pip install triton-windows",
-          "uv pip install https://github.com/woct0rdho/SageAttention/releases/download/v2.1.1-windows/sageattention-2.1.1+cu128torch2.7.0-cp310-cp310-win_amd64.whl", 
+          "uv pip install https://github.com/woct0rdho/SageAttention/releases/download/v2.1.1-windows/sageattention-2.1.1+cu128torch2.7.1-cp310-cp310-win_amd64.whl", 
 //          "uv pip install https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post9/triton-3.2.0-cp310-cp310-win_amd64.whl",
 //          "uv pip install https://github.com/deepbeepmeep/SageAttention/raw/refs/heads/main/releases/sageattention-2.1.0-cp310-cp310-win_amd64.whl"
         ]