From 35623e5da604d806944209a80225258b4bbe9bc3 Mon Sep 17 00:00:00 2001
From: Spiros Maggioros <spirosastro@gmail.com>
Date: Sun, 10 Nov 2024 01:11:25 +0200
Subject: [PATCH] python 3.13 support only for ubuntu | macos build till 3.12

---
 requirements.txt | 2 +-
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index bb0a71a..18e6d12 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-torch --index-url https://download.pytorch.org/whl/nightly/cpu
+torch==2.3.1
 nnunetv2==2.5.1
 argparse
 huggingface_hub
diff --git a/setup.py b/setup.py
index 0018942..f980a5e 100644
--- a/setup.py
+++ b/setup.py
@@ -22,7 +22,7 @@
     packages=find_packages(exclude=["tests", ".github"]),
     python_requires=">=3.8",
     install_requires=[
-        "torch",
+        "torch<=2.3.1",
         "nnunetv2",
         "argparse",
         "huggingface_hub"