xdecoder/Instruct-X-Decoder

1_model_entrypoints = {}2 3def register_model(fn):4    module_name_split = fn.__module__.split('.')5    model_name = module_name_split[-1]6    _model_entrypoints[model_name] = fn7    return fn8 9def model_entrypoints(model_name):10    return _model_entrypoints[model_name]11 12def is_model(model_name):13    return model_name in _model_entrypoints