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,