Upload chatterbox_utils.py with huggingface_hub

chatterbox_utils.py

@@ -118,7 +118,7 @@ def load_chatterbox(device="cuda"):
     if SESSIONS["speech_encoder"]: return
     print(f"🚀 Loading Chatterbox ONNX into {device}...")
     opts = onnxruntime.SessionOptions()
-    provs = ["CUDAExecutionProvider"] if device == "cuda" and onnxruntime.get_device() == "GPU" else ["CPUExecutionProvider"]
+    provs = ["CUDAExecutionProvider", "CPUExecutionProvider"] if device == "cuda" else ["CPUExecutionProvider"]
     
     for sess_name in ["speech_encoder", "embed_tokens", "conditional_decoder", "language_model"]:
         fname = "onnx/" + (sess_name + ".onnx" if sess_name != "language_model" else "language_model.onnx")