diff --git a/hpvm/projects/torch2hpvm/setup.py b/hpvm/projects/torch2hpvm/setup.py index ae103a2cdf0c0872278c147ddac5774ce79da452..f0cd851e586cf4d35c856ead11915f97c7654901 100644 --- a/hpvm/projects/torch2hpvm/setup.py +++ b/hpvm/projects/torch2hpvm/setup.py @@ -7,6 +7,8 @@ setup( author="Yuanjing Shi, Yifan Zhao", author_email="ys26@illinois.edu, yifanz16@illinois.edu", packages=["torch2hpvm"], - install_requires=["jinja2>=2.11", "networkx>=2.5", "onnx>=1.8.0", "torch"], + install_requires=[ + "jinja2>=2.11", "networkx>=2.5", "onnx>=1.8.0", "torch", "onnx-simplifier>=0.2.27" + ], entry_points={"console_scripts": ["torch2hpvm=torch2hpvm:main"]}, )