build: 📌 Remove upper bound on PyTorch (#38789)

build: 📌 remove upper bound on torch dependency as issue which originally resulted in the pin has been released in torch 2.7.1
2025-06-12 15:34:13 +01:00
parent eea35a15b0
commit 3542e0b844
2 changed files with 2 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -189,7 +189,7 @@ _deps = [
    "tiktoken",
    "timm<=1.0.11",
    "tokenizers>=0.21,<0.22",
-    "torch>=2.1,<2.7",  # Installing torch 2.7 results in slower compiled LLMs. Pinned while we investigate.
+    "torch>=2.1",
    "torchaudio",
    "torchvision",
    "pyctcdecode>=0.4.0",