diff --git a/requirements/common.txt b/requirements/common.txt index 6e7fd90d02..6a183e09ac 100644 --- a/requirements/common.txt +++ b/requirements/common.txt @@ -32,7 +32,9 @@ 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 2e03df1bdc..eaa11451a2 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -76,7 +76,9 @@ attrs==26.1.0 audioread==3.0.1 # via librosa av==16.1.0 - # via -r requirements/test/rocm.in + # via + # -r requirements/test/../common.txt + # -r requirements/test/rocm.in azure-core==1.39.0 # via # azure-identity @@ -1331,6 +1333,7 @@ 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 37782062ae..7665978c6b 100644 --- a/setup.py +++ b/setup.py @@ -1092,9 +1092,7 @@ 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