From 29df666d44efd6a53f15b3a7c4875b5d9483a9b2 Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Wed, 18 Dec 2024 14:13:09 +0200
Subject: [PATCH] tts : enable "return_tokens" in Python example

ggml-ci
---
 examples/tts/tts-outetts.py | 1 +
 1 file changed, 1 insertion(+)

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,