TheWhisper/pyproject.toml at main · TheStageAI/TheWhisper · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
[build-system]
requires = ["setuptools>=64", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "thestage-speechkit"
version = "0.1.0"
description = "Optimized Whisper models for streaming and on-device use"
readme = "README.md"
requires-python = ">=3.9"
license = { text = "MIT" }
keywords = ["asr", "whisper", "streaming", "mlx", "nvidia", "fastapi"]
classifiers = [
  "Programming Language :: Python :: 3",
  "License :: OSI Approved :: MIT License",
  "Operating System :: OS Independent",
  "Intended Audience :: Developers",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
  "editdistance",
  "librosa",
  "silero-vad",
  "fastapi",
  "uvicorn",
  "python-multipart",
  "sounddevice",
  "httpx",
  "python-dotenv",
]

[project.optional-dependencies]
nvidia = [
  "torch==2.9.0",
  "transformers==4.52.3",
  "torchaudio==2.9.0",
  "tensorrt==10.13.0.35",
  "tensorrt-cu12==10.13.0.35",
  "tensorrt-cu12-bindings==10.13.0.35",
  "tensorrt-cu12-libs==10.13.0.35",
]
apple = [
  "torch==2.7.0",
  "torchaudio==2.7.0",
  "transformers==4.52.3",
  "editdistance",
  "mlx==0.25.2",
  "coremltools==8.3.0",
]

[project.urls]
Homepage = "https://github.com/TheStageAI/TheWhisper"
Source = "https://github.com/TheStageAI/TheWhisper"
Issues = "https://github.com/TheStageAI/TheWhisper/issues"

[tool.setuptools]
include-package-data = true

[tool.setuptools.packages.find]
include = ["thestage_speechkit*"]
exclude = ["examples*", "scripts*", "tests*"]