setup.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. import os
  2. import platform
  3. import sys
  4. import pkg_resources
  5. from setuptools import find_packages, setup
  6. def read_version(fname="whisper/version.py"):
  7. exec(compile(open(fname, encoding="utf-8").read(), fname, "exec"))
  8. return locals()["__version__"]
  9. requirements = []
  10. if sys.platform.startswith("linux") and platform.machine() == "x86_64":
  11. requirements.append("triton==2.0.0")
  12. setup(
  13. name="openai-whisper",
  14. py_modules=["whisper"],
  15. version=read_version(),
  16. description="Robust Speech Recognition via Large-Scale Weak Supervision",
  17. long_description=open("README.md", encoding="utf-8").read(),
  18. long_description_content_type="text/markdown",
  19. readme="README.md",
  20. python_requires=">=3.8",
  21. author="OpenAI",
  22. url="https://github.com/openai/whisper",
  23. license="MIT",
  24. packages=find_packages(exclude=["tests*"]),
  25. install_requires=requirements
  26. + [
  27. str(r)
  28. for r in pkg_resources.parse_requirements(
  29. open(os.path.join(os.path.dirname(__file__), "requirements.txt"))
  30. )
  31. ],
  32. entry_points={
  33. "console_scripts": ["whisper=whisper.transcribe:cli"],
  34. },
  35. include_package_data=True,
  36. extras_require={"dev": ["pytest", "scipy", "black", "flake8", "isort"]},
  37. )