Skip to content

Commit

Permalink
[Misc][CI/Build] Include cv2 via mistral_common[opencv] (#8951)
Browse files Browse the repository at this point in the history
  • Loading branch information
ywang96 authored Sep 30, 2024
1 parent e01ab59 commit b6d7392
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 2 deletions.
2 changes: 1 addition & 1 deletion requirements-common.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -512,7 +512,6 @@ def _read_requirements(filename: str) -> List[str]:
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 {},
Expand Down

0 comments on commit b6d7392

Please sign in to comment.