require nnunetv2==2.5.1

CBICA · Dec 12, 2024 · 02fe863 · 02fe863
1 parent 0306098
commit 02fe863
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -23,7 +23,7 @@
     python_requires=">=3.8",
     install_requires=[
         "torch<=2.2.1",
-        "nnunetv2",
+        "nnunetv2==2.5.1",
         "argparse",
         "huggingface_hub"
     ],