From 7c4232fc8cb6f2fdced486dde3171438ce660b76 Mon Sep 17 00:00:00 2001
From: JimmyPeilinLi <lipeilin@mail.nwpu.edu.cn>
Date: Wed, 29 Apr 2026 16:08:16 +0000
Subject: [PATCH] [build]: pin accelerate-kt torch 2.9.1 dependency

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 913f2ee2bcd..0a888cc5618 100644
--- a/setup.py
+++ b/setup.py
@@ -82,7 +82,7 @@
         "packaging>=20.0",
         "psutil",
         "pyyaml",
-        "torch>=2.10,<2.12",
+        "torch==2.9.1",
         "huggingface_hub>=0.21.0",
         "safetensors>=0.4.3",
     ],