This commit is contained in:
Andres Marafioti
2024-09-05 17:02:34 +02:00
parent 274f27a7a8
commit c88eec7e26

View File

@@ -147,6 +147,9 @@ class ParlerTTSHandler(BaseHandler):
)
def process(self, llm_sentence):
if isinstance(llm_sentence, tuple):
llm_sentence, _ = llm_sentence
console.print(f"[green]ASSISTANT: {llm_sentence}")
nb_tokens = len(self.prompt_tokenizer(llm_sentence).input_ids)