Add debug logging for processor output keys

2025-12-29 16:17:52 +08:00
parent d2022fc825
commit a60810cdca
1 changed files with 7 additions and 0 deletions
@@ -435,6 +435,13 @@ class AIIA_VibeVoice_TTS:
                 return_tensors="pt"
             )
             
+             print(f"[AIIA DEBUG] Processor output keys: {inputs.keys()}")
+             for k, v in inputs.items():
+                 if v is None:
+                     print(f"[AIIA DEBUG] {k} is None")
+                 else:
+                     print(f"[AIIA DEBUG] {k}: {type(v)} - shape: {v.shape if hasattr(v, 'shape') else 'N/A'}")
+
             # Move all tensors to device
             input_args = {
                 "input_ids": inputs["input_ids"].to(device),