TGPro1 commited on
Commit
c6545d5
·
verified ·
1 Parent(s): ba2715c

Upload chatterbox_utils.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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="cuda"):
117
- """Pre-load ONNX sessions"""
118
  if SESSIONS["speech_encoder"]: return
119
- print(f"🚀 Loading Chatterbox ONNX into {device}...")
120
  opts = onnxruntime.SessionOptions()
121
- provs = ["CUDAExecutionProvider", "CPUExecutionProvider"] if device == "cuda" else ["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")
 
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")