[Misc][CI/Build] Include cv2 via mistral_common[opencv] (#8951)

This commit is contained in:
Roger Wang 2024-09-29 21:28:26 -07:00 committed by GitHub
parent e01ab595d8
commit b6d7392579
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 2 deletions

View File

@ -26,7 +26,7 @@ pyzmq
msgspec
gguf == 0.10.0
importlib_metadata
mistral_common >= 1.4.3
mistral_common[opencv] >= 1.4.4
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>=74.1.1; 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

View File

@ -512,7 +512,6 @@ setup(
ext_modules=ext_modules,
extras_require={
"tensorizer": ["tensorizer>=2.9.0"],
"video": ["opencv-python"], # Required for video processing
"audio": ["librosa", "soundfile"] # Required for audio processing
},
cmdclass={"build_ext": cmake_build_ext} if len(ext_modules) > 0 else {},