models/disco_audio/configuration_disco_audio.py

from transformers import PretrainedConfig
from omegaconf import OmegaConf

class DiscoAudioConfig(PretrainedConfig):
    model_type = "disco_audio"

    def __init__(self, config_obj=None, **kwargs):
        if config_obj is not None:
            cfg_dict = OmegaConf.to_container(config_obj, resolve=True)
            kwargs.update(cfg_dict)

        super().__init__(**kwargs)