Make requirement just torch

CBICA · Nov 6, 2024 · 7feffd2 · 7feffd2
1 parent 4653034
commit 7feffd2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -22,7 +22,7 @@
     packages=find_packages(exclude=["tests", ".github"]),
     python_requires=">=3.9",
     install_requires=[
-        "torch<=2.5.1",
+        "torch",
         "nnunetv2<=2.5.1",
         "argparse",
         "huggingface_hub"