diff --git a/examples/tts/tts-outetts.py b/examples/tts/tts-outetts.py
index 3c085dd5437de..bebd38ef4a5fe 100644
--- 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,