From 415ac32256d58d855227be2acc70f25e72a8467e Mon Sep 17 00:00:00 2001 From: CN_ChiTu <36254426+CNChTu@users.noreply.github.com> Date: Sun, 17 Dec 2023 01:20:31 +0800 Subject: [PATCH] Fix requires --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index dde39a7..2772db1 100644 --- a/setup.py +++ b/setup.py @@ -12,7 +12,7 @@ author='CNChTu', author_email='2921046558@qq.com', url='https://github.com/CNChTu/FCPE', - install_requires=['einops', 'local_attention', 'torchaudio', 'numpy'], + install_requires=['einops', 'local_attention', 'torch', 'torchaudio', 'numpy'], packages=['torchfcpe'], package_data={'torchfcpe': ['assets/*']}, long_description=long_description,