List torch as a installation requirements.

This commit is contained in:
Fangjun Kuang 2022-07-11 09:27:45 +08:00
parent e15b2ae6f5
commit c3c415fb73

View File

@ -41,6 +41,7 @@ setuptools.setup(
long_description_content_type="text/markdown",
ext_modules=[cmake_extension("_kaldifeat")],
cmdclass={"build_ext": BuildExtension, "bdist_wheel": bdist_wheel},
install_requires=["torch"],
zip_safe=False,
classifiers=[
"Programming Language :: C++",