[Misc] Optional installation of audio related packages (#8063)

2024-09-01 14:46:57 -07:00
parent 5231f0898e
commit 5b86b19954
6 changed files with 29 additions and 10 deletions
--- a/setup.py
+++ b/setup.py
@@ -501,6 +501,7 @@ setup(
    ext_modules=ext_modules,
    extras_require={
        "tensorizer": ["tensorizer>=2.9.0"],
+        "audio": ["librosa", "soundfile"]  # Required for audio processing
    },
    cmdclass={"build_ext": cmake_build_ext} if len(ext_modules) > 0 else {},
    package_data=package_data,