From 7e37bc5d9a69e65a1a8b1f3a0b432ebac100a2a5 Mon Sep 17 00:00:00 2001 From: Alice Yanagi Date: Wed, 4 Jun 2025 11:18:00 +0800 Subject: [PATCH] Fix the duration computation in `triton_trtllm/client_grpc.py` (#1071) * Update client_grpc.py Using `actual_duration` to compute metrics like RTF. --- src/f5_tts/runtime/triton_trtllm/client_grpc.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/f5_tts/runtime/triton_trtllm/client_grpc.py b/src/f5_tts/runtime/triton_trtllm/client_grpc.py index 88433b6..5a4adc5 100644 --- a/src/f5_tts/runtime/triton_trtllm/client_grpc.py +++ b/src/f5_tts/runtime/triton_trtllm/client_grpc.py @@ -310,8 +310,9 @@ async def send( audio_save_path = os.path.join(audio_save_dir, f"{item['target_audio_path']}.wav") sf.write(audio_save_path, audio, save_sample_rate, "PCM_16") - latency_data.append((end, estimated_target_duration)) - total_duration += estimated_target_duration + actual_duration = len(audio) / save_sample_rate + latency_data.append((end, actual_duration)) + total_duration += actual_duration return total_duration, latency_data