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 .`
This commit is contained in:
committed by
YaoyaoChang
parent
c1c5e40bef
commit
9b06438560
@@ -0,0 +1,16 @@
|
|||||||
|
# vibevoice/__init__.py
|
||||||
|
from vibevoice.modular import (
|
||||||
|
VibeVoiceStreamingForConditionalGenerationInference,
|
||||||
|
VibeVoiceStreamingConfig,
|
||||||
|
)
|
||||||
|
from vibevoice.processor import (
|
||||||
|
VibeVoiceStreamingProcessor,
|
||||||
|
VibeVoiceTokenizerProcessor,
|
||||||
|
)
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
"VibeVoiceStreamingForConditionalGenerationInference",
|
||||||
|
"VibeVoiceStreamingConfig",
|
||||||
|
"VibeVoiceStreamingProcessor",
|
||||||
|
"VibeVoiceTokenizerProcessor",
|
||||||
|
]
|
||||||
@@ -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",
|
||||||
|
]
|
||||||
@@ -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",
|
||||||
|
]
|
||||||
Reference in New Issue
Block a user