From 83072f76fe5cf4585ed223cbc8cc672116098dee Mon Sep 17 00:00:00 2001 From: Hawk Lee Date: Mon, 29 Dec 2025 16:53:33 +0800 Subject: [PATCH] Pass input_ids positionally to fix transformers generation error --- aiia_vibevoice_nodes.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/aiia_vibevoice_nodes.py b/aiia_vibevoice_nodes.py index 08bc254..59052b4 100644 --- a/aiia_vibevoice_nodes.py +++ b/aiia_vibevoice_nodes.py @@ -443,7 +443,11 @@ class AIIA_VibeVoice_TTS: print(f"[AIIA DEBUG] {k}: {type(v)} - shape: {v.shape if hasattr(v, 'shape') else 'N/A'}") # Move all tensors to device - input_args = {k: v.to(device) if hasattr(v, "to") else v for k, v in inputs.items()} + # Extract input_ids for positional argument + input_ids = inputs["input_ids"].to(device) + + # Prepare other kwargs + input_args = {k: v.to(device) if hasattr(v, "to") else v for k, v in inputs.items() if k != "input_ids"} max_new_tokens = 4096 # Default safe limit @@ -461,6 +465,7 @@ class AIIA_VibeVoice_TTS: with torch.no_grad(): output_wav = model.generate( + input_ids, **input_args, **generation_kwargs )