Fixing setup.py

CBICA · Jan 10, 2025 · fb57fb3 · fb57fb3
1 parent 320834d
commit fb57fb3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -25,7 +25,12 @@
     url="https://github.com/CBICA/DLMUSE/",
     packages=find_packages(exclude=["tests", ".github"]),
     python_requires=">=3.9",
-    install_requires=required,
+    install_requires=[
+        "torch<=2.3.1",
+        "nnunetv2<=2.5.1",
+        "argparse",
+        "huggingface_hub"
+    ],
     entry_points={"console_scripts": ["DLMUSE = DLMUSE.__main__:main"]},
     classifiers=[
         "Intended Audience :: Developers",