Fix audio sample rate access in end2end.py for correct output retrieval

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
tzhouam committed 2026-02-02 08:38:24 +00:00
1 parent 08fa85a1ad
commit 39f7cade27
1 file changed
+1 -1
@@ -246,7 +246,7 @@ def main(args):
request_id = output.request_id
audio_tensor = output.outputs[0].multimodal_output["audio"]
output_wav = os.path.join(output_dir, f"output_{request_id}.wav")
audio_samplerate = output.multimodal_output["sr"].item()
audio_samplerate = output.outputs[0].multimodal_output["sr"].item()
# Convert to numpy array and ensure correct format
audio_numpy = audio_tensor.float().detach().cpu().numpy()