mlx-examples/whisper/setup.py
James Zhao bf921afcbe
Make sure to import the correct "version" module when installing mlx_whisper and mlx_lm from local source code. (#969)
* Make sure to import the correct "version" module when installing the
mlx_whisper package from local source code.

* Make sure to import the correct "version" module when installing the mlx_lm package from local source code

* fix

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-09-03 13:16:21 -07:00

38 lines
1.0 KiB
Python

# Copyright © 2024 Apple Inc.
import sys
from pathlib import Path
from setuptools import find_namespace_packages, setup
package_dir = Path(__file__).parent / "mlx_whisper"
with open(package_dir / "requirements.txt") as fid:
requirements = [l.strip() for l in fid.readlines()]
sys.path.append(str(package_dir))
from _version import __version__
setup(
name="mlx-whisper",
version=__version__,
description="OpenAI Whisper on Apple silicon with MLX and the Hugging Face Hub",
long_description=open("README.md", encoding="utf-8").read(),
long_description_content_type="text/markdown",
readme="README.md",
author_email="mlx@group.apple.com",
author="MLX Contributors",
url="https://github.com/ml-explore/mlx-examples",
license="MIT",
install_requires=requirements,
packages=find_namespace_packages(),
include_package_data=True,
python_requires=">=3.8",
entry_points={
"console_scripts": [
"mlx_whisper = mlx_whisper.cli:main",
]
},
)