Update orpheus-tts/kartoffel_decoder.py

orpheus-tts/kartoffel_decoder.py

@@ -38,13 +38,26 @@ def redistribute_codes_kartoffel(code_list):
     for i in range(num_groups):
         base_idx = 7 * i
         try:
-            layer_1.append(code_list[base_idx])
-            layer_2.append(code_list[base_idx + 1] - 4096)
-            layer_3.append(code_list[base_idx + 2] - (2 * 4096))
-            layer_3.append(code_list[base_idx + 3] - (3 * 4096))
-            layer_2.append(code_list[base_idx + 4] - (4 * 4096))
-            layer_3.append(code_list[base_idx + 5] - (5 * 4096))
-            layer_3.append(code_list[base_idx + 6] - (6 * 4096))
+            # Debug: Zeige die Code-Werte
+            if i == 0:  # Nur für die erste Gruppe
+                print(f"DEBUG KARTOFFEL: First group codes: {code_list[base_idx:base_idx+7]}")
+            
+            # SNAC erwartet Codes im Bereich 0-4095
+            code_0 = min(max(code_list[base_idx], 0), 4095)
+            code_1 = min(max(code_list[base_idx + 1] - 4096, 0), 4095)
+            code_2 = min(max(code_list[base_idx + 2] - (2 * 4096), 0), 4095)
+            code_3 = min(max(code_list[base_idx + 3] - (3 * 4096), 0), 4095)
+            code_4 = min(max(code_list[base_idx + 4] - (4 * 4096), 0), 4095)
+            code_5 = min(max(code_list[base_idx + 5] - (5 * 4096), 0), 4095)
+            code_6 = min(max(code_list[base_idx + 6] - (6 * 4096), 0), 4095)
+            
+            layer_1.append(code_0)
+            layer_2.append(code_1)
+            layer_3.append(code_2)
+            layer_3.append(code_3)
+            layer_2.append(code_4)
+            layer_3.append(code_5)
+            layer_3.append(code_6)
         except IndexError:
             print(f"DEBUG KARTOFFEL: IndexError during code redistribution at group {i}. Skipping group.")
             break