Upload __init__.py

omnivoice/__init__.py

@@ -0,0 +1,28 @@
+import warnings
+from importlib.metadata import PackageNotFoundError, version
+
+warnings.filterwarnings("ignore", module="torchaudio")
+warnings.filterwarnings(
+    "ignore",
+    category=SyntaxWarning,
+    message="invalid escape sequence",
+    module="pydub.utils",
+)
+warnings.filterwarnings(
+    "ignore",
+    category=FutureWarning,
+    module="torch.distributed.algorithms.ddp_comm_hooks",
+)
+
+try:
+    __version__ = version("omnivoice")
+except PackageNotFoundError:
+    __version__ = "0.0.0"
+
+from omnivoice.models.omnivoice import (
+    OmniVoice,
+    OmniVoiceConfig,
+    OmniVoiceGenerationConfig,
+)
+
+__all__ = ["OmniVoice", "OmniVoiceConfig", "OmniVoiceGenerationConfig"]