Update orpheus-tts/decoder.py

orpheus-tts/decoder.py

@@ -94,20 +94,30 @@ def turn_token_into_id(token_string, index):
 async def tokens_decoder(token_gen):
     buffer = []
     count = 0
-    async for token_sim in token_gen:       
+    token_count = 0
+    async for token_sim in token_gen:
+        token_count += 1
+        print(f"DEBUG DECODER: Processing token {token_count}: {repr(token_sim)}")
         token = turn_token_into_id(token_sim, count)
+        print(f"DEBUG DECODER: Converted to ID: {token}")
+        
         if token is None:
             pass
         else:
             if token > 0:
                 buffer.append(token)
                 count += 1
+                print(f"DEBUG DECODER: Added to buffer. Count: {count}, Buffer size: {len(buffer)}")
 
                 if count % 7 == 0 and count > 27:
                     buffer_to_proc = buffer[-28:]
+                    print(f"DEBUG DECODER: Converting buffer to audio. Buffer: {buffer_to_proc}")
                     audio_samples = convert_to_audio(buffer_to_proc, count)
                     if audio_samples is not None:
+                        print(f"DEBUG DECODER: Generated audio chunk of {len(audio_samples)} bytes")
                         yield audio_samples
+                    else:
+                        print("DEBUG DECODER: convert_to_audio returned None")
 
 
 # ------------------ Synchronous Tokens Decoder Wrapper ------------------ #