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