OpenCompass/opencompass/models/ola/model/speech_projector/builder.py
2025-03-04 23:10:00 +08:00

12 lines
366 B
Python

from .speech_projector import EncoderProjectorConcat
def build_speech_projector(config):
projector_type = getattr(config, 'speech_projector_type', 'linear')
if projector_type == 'linear':
return EncoderProjectorConcat(config)
elif projector_type == 'none':
return None
raise ValueError(f'Unknown projector type: {projector_type}')