From 9b0643856052c42a008812c48913f127881278d2 Mon Sep 17 00:00:00 2001
From: RaihanulHaque <raihanulhaque007@gmail.com>
Date: Tue, 9 Dec 2025 00:03:56 +0600
Subject: [PATCH] feat: add __init__.py files to enable module imports

Add __init__.py files to vibevoice/modular and vibevoice/processor
directories to properly export classes and enable package imports.

This allows users to import the package after installation:
- from vibevoice import VibeVoiceStreamingForConditionalGenerationInference
- from vibevoice.modular import VibeVoiceStreamingConfig
- from vibevoice.processor import VibeVoiceStreamingProcessor

Fixes import errors when using `pip install -e .`
---
 vibevoice/__init__.py           | 16 ++++++++++++++++
 vibevoice/modular/__init__.py   | 14 ++++++++++++++
 vibevoice/processor/__init__.py | 11 +++++++++++
 3 files changed, 41 insertions(+)

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