From 4647f05d066653f34dcecd49c50118aaa06689f2 Mon Sep 17 00:00:00 2001
From: Spiros Maggioros <spirosastro@gmail.com>
Date: Sun, 10 Nov 2024 01:50:02 +0200
Subject: [PATCH] Can't install torch==2.2.1 with nightly version, on hold till
 they fully support it

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

diff --git a/setup.py b/setup.py
index aabe76d..df2e5d8 100644
--- a/setup.py
+++ b/setup.py
@@ -22,7 +22,7 @@
     packages=find_packages(exclude=["tests", ".github"]),
     python_requires=">=3.9",
     install_requires=[
-        "torch",
+        "torch<=2.2.1",
         "nnunetv2<=2.2.1",
         "argparse",
         "huggingface_hub"