[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:
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
|
||||
|
||||
Reference in new issue
Block a user