Fixed torch version

CBICA · Nov 9, 2024 · ce7a571 · ce7a571
1 parent 35623e5
commit ce7a571
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-torch==2.3.1
+torch==2.2.1
 nnunetv2==2.5.1
 argparse
 huggingface_hub

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