diff --git a/requirements/common.txt b/requirements/common.txt index 6a183e09acf..6e7fd90d023 100644 --- a/requirements/common.txt +++ b/requirements/common.txt @@ -32,9 +32,7 @@ pyzmq >= 25.0.0 msgspec gguf >= 0.17.0 mistral_common[image] >= 1.11.0 -av # required for audio in video IO opencv-python-headless >= 4.13.0 # required for video IO -soundfile # required for audio IO pyyaml six>=1.16.0; python_version > '3.11' # transitive dependency of pandas that needs to be the latest version for python 3.12 setuptools>=77.0.3,<81.0.0; python_version > '3.11' # Setuptools is used by triton, we need to ensure a modern version is installed for 3.12+ so that it does not try to import distutils, which was removed in 3.12 diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index 5abec4dce4c..33ea3a3f0db 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -76,9 +76,7 @@ attrs==26.1.0 audioread==3.0.1 # via librosa av==16.1.0 - # via - # -r requirements/test/../common.txt - # -r requirements/test/rocm.in + # via -r requirements/test/rocm.in azure-core==1.39.0 # via # azure-identity @@ -1333,7 +1331,6 @@ sortedcontainers==2.4.0 # via hypothesis soundfile==0.13.1 # via - # -r requirements/test/../common.txt # -r requirements/test/rocm.in # genai-perf # librosa diff --git a/setup.py b/setup.py index f6276616a19..bb2d6ac545d 100644 --- a/setup.py +++ b/setup.py @@ -1094,7 +1094,9 @@ setup( "instanttensor": ["instanttensor >= 0.1.5"], "runai": ["runai-model-streamer[s3,gcs,azure] >= 0.15.7"], "audio": [ + "av", "scipy", + "soundfile", "mistral_common[audio]", ], # Required for audio processing "video": [], # Kept for backwards compatibility