forked from devnen/Chatterbox-TTS-Server
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements-nvidia-cu128.txt
More file actions
66 lines (59 loc) · 2.15 KB
/
Copy pathrequirements-nvidia-cu128.txt
File metadata and controls
66 lines (59 loc) · 2.15 KB
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
62
63
64
65
66
# ============================================================================
# CHATTERBOX TTS SERVER - REQUIREMENTS (NVIDIA CUDA 12.8 / BLACKWELL)
# ============================================================================
# This file provides CUDA 12.8 support for Blackwell architecture (sm_120):
# - RTX 5060 Ti, RTX 5070, RTX 5070 Ti, RTX 5080, RTX 5090
#
# For RTX 20/30/40 series GPUs, use requirements-nvidia.txt instead.
#
# ============================================================================
# IMPORTANT: chatterbox-tts is NOT included here. It must be installed
# separately with --no-deps to prevent PyTorch from being downgraded.
#
# The start.py launcher handles this automatically.
#
# For manual installation, run these commands in order:
# pip install -r requirements-nvidia-cu128.txt
# pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0
# ============================================================================
# --- PyTorch CUDA 12.8 Package Index ---
--extra-index-url https://download.pytorch.org/whl/cu128
# --- Deep Learning Framework (CUDA 12.8 with sm_120 Blackwell support) ---
torch==2.9.0
torchvision==0.24.0
torchaudio==2.9.0
# --- Chatterbox Dependencies (explicit, since --no-deps skips them) ---
# NOTE: s3tokenizer and onnx are installed with --no-deps alongside chatterbox
# to avoid protobuf version conflicts (onnx requires protobuf>=3.20.2 but
# descript-audiotools requires protobuf<3.20).
conformer==0.3.2
diffusers==0.29.0
transformers==4.46.3
einops>=0.7.0
omegaconf==2.3.0
resampy==0.4.3
resemble-perth==1.0.1
# --- Core Web Framework ---
fastapi>=0.100.0,<0.116.0 # starlette<1.0 needed for TemplateResponse compat
uvicorn[standard]>=0.27.0
# --- ML & Audio Libraries ---
numpy>=1.24.0
soundfile>=0.12.0
librosa>=0.10.0
safetensors>=0.4.0
descript-audio-codec==1.0.0
# --- Configuration & Utilities ---
PyYAML>=6.0
python-multipart>=0.0.5
requests>=2.28.0
Jinja2>=3.1.2,<3.2.0
watchdog>=3.0.0
aiofiles>=23.0.0
unidecode>=1.3.0
inflect>=7.0.0
tqdm>=4.65.0
hf_transfer>=0.1.4
# --- Audio Post-processing ---
pydub>=0.25.0
audiotsm>=0.1.2
praat-parselmouth>=0.4.0