Spaces:

xdecoder
/

Demo

Runtime error

App Files Files Community

Demo / xdecoder /body /build.py

MaureenZOU

init

e972e1f over 1 year ago

raw history blame

No virus

365 Bytes

	from .registry import model_entrypoints
	from .registry import is_model

	from .xdecoder_head import *


	def build_xdecoder_head(config, args, *kwargs):
	model_name = config['MODEL']['HEAD']
	if not is_model(model_name):
	raise ValueError(f'Unkown model: {model_name}')

	body = model_entrypoints(model_name)(config, args, *kwargs)
	return body