[Voxtral TTS] Correct decode steps param in Voxtral TTS (#2524)

Signed-off-by: Chen-Yo Sun <chenyo.sun@mistral.ai>
This commit is contained in:
Chen-Yo Sun authored and GitHub committed 2026-04-13 17:53:35 +00:00
1 parent 6b5a52ae31
commit c9e2e3e8d7
4 files changed
+22 -5

No files matched your search

@@ -78,6 +78,13 @@ except Exception:
AudioSpecialTokens = _mod2.AudioSpecialTokens
class SyntheticAcousticTransformerArgs:
"""Mimics AcousticTransformerArgs interface."""
def __init__(self):
self.n_decoding_steps = 7
class SyntheticModelArgs:
"""Mimics MultimodalAudioModelArgs interface."""
@@ -96,6 +103,7 @@ class SyntheticAcousticTransformer(nn.Module):
def __init__(self):
super().__init__()
self.model_args = SyntheticModelArgs()
self.acoustic_transformer_args = SyntheticAcousticTransformerArgs()
self.acoustic_embeddings_levels = ACOUSTIC_EMBEDDINGS_LEVELS
# semantic_codebook_output: hidden_dim -> padded_codebook_size