Add Chatterbox Turbo HF wrapper

README.md

@@ -0,0 +1,16 @@
+---
+pipeline_tag: text-to-speech
+license: apache-2.0
+---
+
+# Chatterbox Turbo (HF Wrapper)
+
+This repository wraps ResembleAI Chatterbox Turbo in a
+Transformers-compatible text-to-speech pipeline.
+
+## Usage
+
+Input: text  
+Output: audio waveform
+
+Supports zero-shot voice cloning via voice embeddings.

__init__.py.txt

@@ -0,0 +1,7 @@
+from transformers import PIPELINE_REGISTRY
+from .pipeline import ChatterboxTurboPipeline
+
+PIPELINE_REGISTRY.register_pipeline(
+    "text-to-speech",
+    pipeline_class=ChatterboxTurboPipeline
+)

config.json

@@ -0,0 +1,4 @@
+{
+  "model_type": "chatterbox-turbo",
+  "architectures": ["ChatterboxTurboPipeline"]
+}

pipeline.py.txt

@@ -0,0 +1,14 @@
+from transformers import Pipeline
+from chatterbox.turbo import synthesize_turbo
+
+class ChatterboxTurboPipeline(Pipeline):
+    def _sanitize_parameters(self, **kwargs):
+        return {}, {}, kwargs
+
+    def __call__(self, text, voice_embedding=None, **kwargs):
+        audio = synthesize_turbo(
+            text=text,
+            voice_embedding=voice_embedding,
+            **kwargs
+        )
+        return audio

requirements.txt

@@ -0,0 +1,5 @@
+torch
+transformers
+soundfile
+numpy
+git+https://github.com/resemble-ai/chatterbox.git