diff --git a/alltalk.py b/alltalk.py index 4217648..5e7b4bb 100644 --- a/alltalk.py +++ b/alltalk.py @@ -391,8 +391,10 @@ class AllTalkTTSNode: ) # Convert numpy array to PyTorch tensor (required by ComfyUI) + # ComfyUI's AudioSaveHelper expects shape (samples, channels) + # but our parsers return (channels, samples), so we need to transpose import torch - audio_tensor = torch.from_numpy(audio_data) + audio_tensor = torch.from_numpy(audio_data.T) audio_dict = { "waveform": audio_tensor,