Team Ai
Apppublic

xdecoder/Instruct-X-Decoder

sourceHugging Faceafl-3.0updated 3y agoView on Hugging Face
163likes
build.py12 linesDownload Raw Back to decoder
1from .registry import model_entrypoints2from .registry import is_model3 4from .xdecoder import *5 6def build_decoder(config, *args, **kwargs):7    model_name = config['MODEL']['DECODER']['NAME']8 9    if not is_model(model_name):10        raise ValueError(f'Unkown model: {model_name}')11 12    return model_entrypoints(model_name)(config, *args, **kwargs)