tts : enable "return_tokens" in Python example

ggml-ci
ggerganov · Dec 18, 2024 · 29df666 · 29df666
1 parent 2a1a6f6
commit 29df666
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/tts/tts-outetts.py b/examples/tts/tts-outetts.py
@@ -132,6 +132,7 @@ def process_text(text: str):
         "prompt": [prefix + words, *suffix],
         "n_predict": 1024,
         "cache_prompt": True,
+        "return_tokens": True,
         "samplers": ["top_k"],
         "top_k": 16,
         "seed": 1003,