require nnunetv2==2.5.1

spirosmaggioros · spirosmaggioros · commit 02fe863a2060 · 2024-12-12T02:11:58.000+02:00
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"
     ],
