mirror of
https://github.com/UX-Decoder/Segment-Everything-Everywhere-All-At-Once.git
synced 2025-06-03 14:50:11 +08:00
11 lines
317 B
Python
11 lines
317 B
Python
|
from .registry import model_entrypoints
|
||
|
|
from .registry import is_model
|
||
|
|
|
||
|
|
|
||
|
|
def build_language_encoder(config, **kwargs):
|
||
|
|
model_name = config['MODEL']['TEXT']['ARCH']
|
||
|
|
|
||
|
|
if not is_model(model_name):
|
||
|
|
raise ValueError(f'Unkown model: {model_name}')
|
||
|
|
|
||
|
|
return model_entrypoints(model_name)(config, **kwargs)
|