This commit is contained in:
XXXXRT666 2025-10-11 18:06:14 +01:00
parent 8f7e5557be
commit 915ed53c76

View File

@ -959,14 +959,17 @@ def get_tts_wav(
console.print(f">> Time Stamps: {t0:.3f}\t{t1:.3f}\t{t2:.3f}\t{t3:.3f}")
console.print(f">> Infer Speed: {infer_speed_avg:.2f} Token/s")
console.print(f">> RTF: {rtf_value:.2f}")
if ttfb_time > 2:
console.print(f">> TTFB: {ttfb_time:.3f} s")
else:
console.print(f">> TTFB: {ttfb_time * 1000:.3f} ms")
gr.Info(f"{infer_speed_avg:.2f} Token/s", title="Infer Speed")
gr.Info(f"{rtf_value:.2f}", title="RTF")
if ttfb_time > 2:
console.print(f">> TTFB: {ttfb_time:.3f} s")
gr.Info(f">> TTFB: {ttfb_time:.3f} s")
else:
console.print(f">> TTFB: {ttfb_time * 1000:.3f} ms")
gr.Info(f">> TTFB: {ttfb_time * 1000:.3f} ms")
yield opt_sr, (audio_opt_n * 32767).astype(np.int16)
if torch.cuda.is_available():