tts support latest transformers(4.57.6)

This commit is contained in:
YaoyaoChang
2026-01-26 03:14:07 -08:00
parent c4ee4fe716
commit a00f431e14
2 changed files with 202 additions and 14 deletions
@@ -81,6 +81,15 @@ class VibeVoiceStreamingConfig(PretrainedConfig):
super().__init__(**kwargs)
def get_text_config(self, decoder=False):
"""Returns the decoder config (required for transformers >= 4.57 cache compatibility)."""
return self.decoder_config
@property
def num_hidden_layers(self):
"""Proxy to decoder_config.num_hidden_layers (required for transformers >= 4.57)."""
return self.decoder_config.num_hidden_layers
def to_dict(self):
"""
Override to_dict to handle torch.dtype serialization.