diff --git a/vibevoice/__init__.py b/vibevoice/__init__.py index e69de29..b00e91a 100644 --- a/vibevoice/__init__.py +++ b/vibevoice/__init__.py @@ -0,0 +1,16 @@ +# vibevoice/__init__.py +from vibevoice.modular import ( + VibeVoiceStreamingForConditionalGenerationInference, + VibeVoiceStreamingConfig, +) +from vibevoice.processor import ( + VibeVoiceStreamingProcessor, + VibeVoiceTokenizerProcessor, +) + +__all__ = [ + "VibeVoiceStreamingForConditionalGenerationInference", + "VibeVoiceStreamingConfig", + "VibeVoiceStreamingProcessor", + "VibeVoiceTokenizerProcessor", +] \ No newline at end of file diff --git a/vibevoice/modular/__init__.py b/vibevoice/modular/__init__.py index e69de29..c92b1a7 100644 --- a/vibevoice/modular/__init__.py +++ b/vibevoice/modular/__init__.py @@ -0,0 +1,14 @@ +# vibevoice/modular/__init__.py +from .modeling_vibevoice_streaming_inference import VibeVoiceStreamingForConditionalGenerationInference +from .configuration_vibevoice_streaming import VibeVoiceStreamingConfig +from .modeling_vibevoice_streaming import VibeVoiceStreamingModel, VibeVoiceStreamingPreTrainedModel +from .streamer import AudioStreamer, AsyncAudioStreamer + +__all__ = [ + "VibeVoiceStreamingForConditionalGenerationInference", + "VibeVoiceStreamingConfig", + "VibeVoiceStreamingModel", + "VibeVoiceStreamingPreTrainedModel", + "AudioStreamer", + "AsyncAudioStreamer", +] \ No newline at end of file diff --git a/vibevoice/processor/__init__.py b/vibevoice/processor/__init__.py index e69de29..c945e6d 100644 --- a/vibevoice/processor/__init__.py +++ b/vibevoice/processor/__init__.py @@ -0,0 +1,11 @@ +# vibevoice/processor/__init__.py +from .vibevoice_processor import VibeVoiceProcessor +from .vibevoice_streaming_processor import VibeVoiceStreamingProcessor +from .vibevoice_tokenizer_processor import VibeVoiceTokenizerProcessor, AudioNormalizer + +__all__ = [ + "VibeVoiceProcessor", + "VibeVoiceStreamingProcessor", + "VibeVoiceTokenizerProcessor", + "AudioNormalizer", +] \ No newline at end of file