xdecoder/Instruct-X-Decoder

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