models/init.py

"""
Music generation models package
"""
from .musicgen_model import MusicGenModel
from .audioldm_model import AudioLDMModel
from .model_manager import ModelManager

__all__ = ["MusicGenModel", "AudioLDMModel", "ModelManager"]