We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5448450 commit e80127dCopy full SHA for e80127d
setup.py
@@ -5,8 +5,8 @@
5
with open("README.md", "r", encoding="utf-8") as fh:
6
long_description = fh.read()
7
8
-with open("requirements.txt", "r", encoding="utf-8") as fh:
9
- requirements = [line.strip() for line in fh if line.strip() and not line.startswith("#")]
+# Dependencies
+requirements = ["torch>=2.0", "numpy"]
10
11
setuptools.setup(
12
name="torch-pruning",
0 commit comments