models/disco_audio/__init__.py

from .configuration_disco_audio import DiscoAudioConfig
from .modeling_disco_audio import DiscoAudioModel, DiscoAudioPreTrainedModel

__all__ = [
    "DiscoAudioConfig",
    "DiscoAudioModel",
    "DiscoAudioPreTrainedModel",
]