diff --git a/poetry.lock b/poetry.lock index b58254dea..4a204ac3c 100644 --- a/poetry.lock +++ b/poetry.lock @@ -3768,6 +3768,17 @@ files = [ {file = "monotonic-1.6.tar.gz", hash = "sha256:3a55207bcfed53ddd5c5bae174524062935efed17792e9de2ad0205ce9ad63f7"}, ] +[[package]] +name = "more-itertools" +version = "10.5.0" +description = "More routines for operating on iterables, beyond itertools" +optional = false +python-versions = ">=3.8" +files = [ + {file = "more-itertools-10.5.0.tar.gz", hash = "sha256:5482bfef7849c25dc3c6dd53a6173ae4795da2a41a80faea6700d9f5846c5da6"}, + {file = "more_itertools-10.5.0-py3-none-any.whl", hash = "sha256:037b0d3203ce90cca8ab1defbbdac29d5f993fc20131f3664dc8d6acfa872aef"}, +] + [[package]] name = "moviepy" version = "1.0.3" @@ -4243,6 +4254,28 @@ typing-extensions = ">=4.11,<5" [package.extras] datalib = ["numpy (>=1)", "pandas (>=1.2.3)", "pandas-stubs (>=1.1.0.11)"] +[[package]] +name = "openai-whisper" +version = "20240930" +description = "Robust Speech Recognition via Large-Scale Weak Supervision" +optional = false +python-versions = ">=3.8" +files = [ + {file = "openai-whisper-20240930.tar.gz", hash = "sha256:b7178e9c1615576807a300024f4daa6353f7e1a815dac5e38c33f1ef055dd2d2"}, +] + +[package.dependencies] +more-itertools = "*" +numba = "*" +numpy = "*" +tiktoken = "*" +torch = "*" +tqdm = "*" +triton = {version = ">=2.0.0", markers = "platform_machine == \"x86_64\" and sys_platform == \"linux\" or sys_platform == \"linux2\""} + +[package.extras] +dev = ["black", "flake8", "isort", "pytest", "scipy"] + [[package]] name = "opencv-python" version = "4.10.0.84" @@ -8345,6 +8378,28 @@ torchhub = ["filelock", "huggingface-hub (>=0.14.1,<1.0)", "importlib-metadata", video = ["av (==9.2.0)", "decord (==0.6.0)"] vision = ["Pillow"] +[[package]] +name = "triton" +version = "3.1.0" +description = "A language and compiler for custom Deep Learning operations" +optional = false +python-versions = "*" +files = [ + {file = "triton-3.1.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:6b0dd10a925263abbe9fa37dcde67a5e9b2383fc269fdf59f5657cac38c5d1d8"}, + {file = "triton-3.1.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:0f34f6e7885d1bf0eaaf7ba875a5f0ce6f3c13ba98f9503651c1e6dc6757ed5c"}, + {file = "triton-3.1.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:c8182f42fd8080a7d39d666814fa36c5e30cc00ea7eeeb1a2983dbb4c99a0fdc"}, + {file = "triton-3.1.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:6dadaca7fc24de34e180271b5cf864c16755702e9f63a16f62df714a8099126a"}, + {file = "triton-3.1.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:aafa9a20cd0d9fee523cd4504aa7131807a864cd77dcf6efe7e981f18b8c6c11"}, +] + +[package.dependencies] +filelock = "*" + +[package.extras] +build = ["cmake (>=3.20)", "lit"] +tests = ["autopep8", "flake8", "isort", "llnl-hatchet", "numpy", "pytest", "scipy (>=1.7.1)"] +tutorials = ["matplotlib", "pandas", "tabulate"] + [[package]] name = "twisted" version = "24.10.0" @@ -9174,4 +9229,4 @@ testing = ["coverage[toml]", "zope.event", "zope.testing"] [metadata] lock-version = "2.0" python-versions = ">=3.10,<3.14" -content-hash = "9a2eee9b8b5c894c3e6080044fb6942bd179602edf728c9f216b1c0cf4ba45ac" +content-hash = "9b98c96a50451de3edfedc5627e895a74c84f6af3af535fee09a343806e3fbdf" diff --git a/pyproject.toml b/pyproject.toml index 97f9c1cd2..ce1c617b8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -111,6 +111,7 @@ llvmlite = "^0.43.0" ell-ai = "^0.0.14" pynput = "^1.7.7" multiprocessing-utils = "^0.4" +openai-whisper = "^20240930" [tool.pytest.ini_options] filterwarnings = [ # suppress warnings starting from "setuptools>=67.3" diff --git a/scripts/postinstall.py b/scripts/postinstall.py index 417c772f6..061f0c7aa 100755 --- a/scripts/postinstall.py +++ b/scripts/postinstall.py @@ -23,24 +23,6 @@ def install_detectron2() -> None: sys.exit(1) -def install_whisper() -> None: - """Install Whisper from its GitHub repository.""" - try: - subprocess.check_call( - [ - sys.executable, - "-m", - "pip", - "install", - "git+https://github.com/openai/whisper.git", - "--no-build-isolation", - ] - ) - except subprocess.CalledProcessError as e: - print(f"Error installing Whisper: {e}") - sys.exit(1) - - def install_dashboard() -> None: """Install dashboard dependencies based on the operating system.""" original_directory = os.getcwd() @@ -82,7 +64,6 @@ def main() -> None: """Main function to install dependencies.""" try: install_detectron2() - install_whisper() install_dashboard() except Exception as e: print(f"Unhandled error: {e}")