Upload chatterbox_utils.py with huggingface_hub
Browse files- chatterbox_utils.py +4 -4
chatterbox_utils.py
CHANGED
|
@@ -113,12 +113,12 @@ def prepare_language(txt, lang_id):
|
|
| 113 |
elif lang_id == 'ko': txt = korean_normalize(txt)
|
| 114 |
return f"[{lang_id.lower()}]{txt}" if lang_id else txt
|
| 115 |
|
| 116 |
-
def load_chatterbox(device="
|
| 117 |
-
"""Pre-load ONNX sessions"""
|
| 118 |
if SESSIONS["speech_encoder"]: return
|
| 119 |
-
print(f"🚀 Loading Chatterbox ONNX into
|
| 120 |
opts = onnxruntime.SessionOptions()
|
| 121 |
-
provs = ["
|
| 122 |
|
| 123 |
for sess_name in ["speech_encoder", "embed_tokens", "conditional_decoder", "language_model"]:
|
| 124 |
fname = "onnx/" + (sess_name + ".onnx" if sess_name != "language_model" else "language_model.onnx")
|
|
|
|
| 113 |
elif lang_id == 'ko': txt = korean_normalize(txt)
|
| 114 |
return f"[{lang_id.lower()}]{txt}" if lang_id else txt
|
| 115 |
|
| 116 |
+
def load_chatterbox(device="cpu"):
|
| 117 |
+
"""Pre-load ONNX sessions - v111: Forced CPU for stability"""
|
| 118 |
if SESSIONS["speech_encoder"]: return
|
| 119 |
+
print(f"🚀 Loading Chatterbox ONNX into CPU (ZeroGPU Safe Mode)...")
|
| 120 |
opts = onnxruntime.SessionOptions()
|
| 121 |
+
provs = ["CPUExecutionProvider"]
|
| 122 |
|
| 123 |
for sess_name in ["speech_encoder", "embed_tokens", "conditional_decoder", "language_model"]:
|
| 124 |
fname = "onnx/" + (sess_name + ".onnx" if sess_name != "language_model" else "language_model.onnx")
|