limit vits for one input a time

This commit is contained in:
zpeng11 2025-08-20 18:39:21 -04:00
parent fd0fb35a49
commit 94b31a250f

View File

@ -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,