From 94b31a250f4e50c1b0069e1d1d42ef8402c889ef Mon Sep 17 00:00:00 2001
From: zpeng11 <a2843173630@gmail.com>
Date: Wed, 20 Aug 2025 18:39:21 -0400
Subject: [PATCH] limit vits for one input a time

---
 GPT_SoVITS/onnx_export.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/GPT_SoVITS/onnx_export.py b/GPT_SoVITS/onnx_export.py
index e2aa4ea3..70a5aee7 100644
--- a/GPT_SoVITS/onnx_export.py
+++ b/GPT_SoVITS/onnx_export.py
@@ -259,10 +259,10 @@ class GptSoVits(nn.Module):
             input_names=["text_seq", "pred_semantic", "ref_audio", "spectrum"],
             output_names=["audio"],
             dynamic_axes={
-                "text_seq": {0:"batch_size",1: "text_length"},
-                "pred_semantic": {0: "batch_size", 2: "pred_length"},
-                "ref_audio": {0: "batch_size", 1: "audio_length"},
-                "spectrum": {0: "batch_size", 2: "spectrum_length"},
+                "text_seq": {1: "text_length"},
+                "pred_semantic": {2: "pred_length"},
+                "ref_audio": {1: "audio_length"},
+                "spectrum": {2: "spectrum_length"},
             },
             opset_version=17,
             verbose=False,