Fix requires

CNChTu · Dec 16, 2023 · 415ac32 · 415ac32
1 parent 9d794b4
commit 415ac32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -12,7 +12,7 @@
     author='CNChTu',
     author_email='[email protected]',
     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,