File size: 706 Bytes
e6aed17 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | 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"]
|