tts/model/registry.py

ENCODER_REGISTRY = {}
DECODER_REGISTRY = {}


def register_encoder(name):
    def wrapper(cls):
        ENCODER_REGISTRY[name] = cls
        return cls

    return wrapper


def register_decoder(name):
    def wrapper(cls):
        DECODER_REGISTRY[name] = cls
        return cls

    return wrapper