Correct torch version

CBICA · Dec 11, 2024 · fa8f1c8 · fa8f1c8
1 parent 2a0c6af
commit fa8f1c8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
-torch>=2.2.1
-nnunetv2>=2.2.1
+torch==2.2.1
+nnunetv2==2.5.1
 argparse
 huggingface_hub
 pathlib
diff --git a/setup.py b/setup.py
@@ -20,9 +20,9 @@
     download_url="https://github.com/CBICA/DLWMLS/",
     url="https://github.com/CBICA/DLWMLS/",
     packages=find_packages(exclude=["tests", ".github"]),
-    python_requires=">=3.8",
+    python_requires=">=3.9",
     install_requires=[
-        "torch>=2.2.1",
+        "torch<=2.2.1",
         "nnunetv2>=2.2.1",
         "argparse",
         "huggingface_hub",