Downloading from snapshot

download_model.py

@@ -1,11 +1,9 @@
-import torch
-import functools
-from chatterbox.mtl_tts import ChatterboxMultilingualTTS
+from huggingface_hub import snapshot_download
 
-# Patch torch.load to force CPU mapping during build time
-torch.load = functools.partial(torch.load, map_location='cpu')
-
-print("Downloading model...")
-# Download weights to cache during build time
-ChatterboxMultilingualTTS.from_pretrained(device="cpu")
+print("Downloading model weights (cache only)...")
+# The correct repository for chatterbox-tts
+snapshot_download(
+    repo_id="ResembleAI/chatterbox",
+    allow_patterns=["ve.pt", "t3_mtl23ls_v2.safetensors", "s3gen.pt", "grapheme_mtl_merged_expanded_v1.json", "conds.pt", "Cangjie5_TC.json"]
+)
 print("Model downloaded successfully.")

requirements.txt

@@ -2,6 +2,7 @@ fastapi==0.127.0
 uvicorn==0.40.0
 pydantic==2.11.10
 chatterbox-tts==0.1.6
+peft
 python-multipart==0.0.21
 numpy==1.25.2
 scipy==1.16.3