fixed audio shape issue

2024-10-09 00:07:59 -04:00
parent d57f222fa2
commit c84e7732d3
2 changed files with 5 additions and 8 deletions
@@ -16,7 +16,7 @@ temp_path = folder_paths.get_temp_directory()

 def to_audio_dict(audio: "np.ndarray", sr: int):
    audio = np.atleast_2d(audio)
-    waveform = torch.from_numpy(audio.reshape((-1,audio.shape[0]))).transpose(0,1).unsqueeze(0)
+    waveform = torch.from_numpy(audio.reshape((-1,audio.shape[0]))).unsqueeze(0)
    return dict(waveform=waveform,sample_rate=sr)

 class LoadAudio: