Add streaming config and update model card

README.md

@@ -8,6 +8,8 @@ tags:
   - audio
   - speech-synthesis
   - voice-cloning
+  - streaming-tts
+  - double-streaming
   - autoregressive
   - meanflow
   - two-step
@@ -21,27 +23,54 @@ tags:
   <a href="https://studio-dots-ai.github.io/dots.tts-demo/"><img src="https://img.shields.io/badge/Demo%20Page-Live-red" alt="Demo Page"></a>
 </p>
 
-This repository provides a standalone dots.tts artifact for **fixed two-step STTS inference**.
-The sampling contract is stored in `config.json` and selected automatically.
-Sampling options should be omitted.
+This repository provides a standalone dots.tts artifact for **fixed two-step double-streaming TTS inference**.
+The sampling and streaming contracts are stored in `config.json` and selected automatically.
+Sampling and streaming cadence options should be omitted.
 
 ## Quick start
 
 ```python
-from dots_tts.runtime import DotsTtsRuntime
+import torch
+import soundfile as sf
 
-runtime = DotsTtsRuntime.from_pretrained(
+from dots_tts.runtime_double_streaming import DotsTtsRuntimeDoubleStreaming
+
+runtime = DotsTtsRuntimeDoubleStreaming.from_pretrained(
     "dots-studio/dots.tts-mf-2steps-stts",
     precision="bfloat16",
+    optimize=True,
+    max_generate_length=500,
+)
+
+text = "Hello from two-step double streaming."
+text_token_ids = runtime.model.tokenizer.encode(text, add_special_tokens=False)
+
+session = runtime.start_double_streaming(
+    prompt_audio_path="/path/to/reference.wav",
+    prompt_text="The exact transcript spoken in the reference audio.",
 )
-result = runtime.generate(text="Hello from two-step MeanFlow.")
+
+chunks = []
+for token_id in text_token_ids:
+    chunk = session.push_text_token(token_id)
+    if chunk is not None:
+        chunks.append(chunk.detach().cpu())
+
+for chunk in session.finish_text():
+    chunks.append(chunk.detach().cpu())
+
+audio = torch.cat(chunks, dim=-1).float().squeeze().numpy()
+sf.write("double_streaming.wav", audio, runtime.sample_rate)
 ```
 
-Passing incompatible sampling values raises an error.
+For a complete command-line example, see
+[`scripts/example_double_streaming.py`](https://github.com/studio-dots-ai/dots.tts/blob/main/scripts/example_double_streaming.py).
+
+Passing incompatible sampling values raises an error. Streaming cadence is selected from the artifact configuration.
 
 ## Scope and limitations
 
-This model is intended for fixed two-step inference. Other sampling settings are not claimed.
+This model is intended for fixed two-step double-streaming inference. Other sampling or streaming settings are not claimed.
 High-fidelity voice cloning must be used only with authorization and consent; do not use it
 for impersonation, fraud, or disinformation.
 

config.json

@@ -127,5 +127,11 @@
     "num_steps": 2,
     "guidance_scale": 0.0,
     "tau_mid": 1.3
+  },
+  "streaming": {
+    "interleave_mode": "buffered_ratio",
+    "initial_lookahead": 3,
+    "warmup_ta": 0,
+    "ta_per_tta": 1
   }
 }