Add VibeVoice ASR (#48)

This commit is contained in:
mrfakename
2026-01-21 21:49:08 -08:00
committed by GitHub
parent b9d561240a
commit fcdbfecd7d
9 changed files with 3015 additions and 4 deletions
+11
View File
@@ -12,6 +12,12 @@ from .configuration_vibevoice import (
VibeVoiceAcousticTokenizerConfig,
VibeVoiceSemanticTokenizerConfig,
VibeVoiceDiffusionHeadConfig,
VibeVoiceASRConfig,
)
from .modeling_vibevoice_asr import (
VibeVoiceASRPreTrainedModel,
VibeVoiceASRModel,
VibeVoiceASRForConditionalGeneration,
)
from .modeling_vibevoice import (
VibeVoicePreTrainedModel,
@@ -67,10 +73,15 @@ __all__ = [
"VibeVoiceAcousticTokenizerConfig",
"VibeVoiceSemanticTokenizerConfig",
"VibeVoiceDiffusionHeadConfig",
"VibeVoiceASRConfig",
# Multi-speaker models
"VibeVoicePreTrainedModel",
"VibeVoiceModel",
"VibeVoiceForConditionalGenerationInference",
# ASR models
"VibeVoiceASRPreTrainedModel",
"VibeVoiceASRModel",
"VibeVoiceASRForConditionalGeneration",
# Streaming configs
"VibeVoiceStreamingConfig",
# Streaming models