clean code

2026-02-05 18:09:24 +08:00 · 2025-07-27 23:33:10 -07:00
parent 5427c274e3
commit 178da09993
6 changed files with 23 additions and 29 deletions
--- a/runtime/triton_trtllm/client_http.py
+++ b/runtime/triton_trtllm/client_http.py
@@ -162,8 +162,8 @@ if __name__ == "__main__":
    result = rsp.json()
    audio = result["outputs"][0]["data"]
    audio = np.array(audio, dtype=np.float32)
-    if args.model_name == "cosyvoice2":
-        sample_rate = 24000
-    else:
+    if args.model_name == "spark_tts":
        sample_rate = 16000
+    else:
+        sample_rate = 24000
    sf.write(args.output_audio, audio, sample_rate, "PCM_16")