Spaces:
Running
on
Zero
Running
on
Zero
File size: 438 Bytes
1938217 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from .speech_encoder import WhisperWrappedEncoder, DualWrappedEncoder
def build_speech_encoder(config):
speech_encoder_type = getattr(config, 'speech_encoder_type', None)
if "whisper" in speech_encoder_type.lower():
return WhisperWrappedEncoder.load(config)
elif "dual" in speech_encoder_type.lower():
return DualWrappedEncoder(config)
raise ValueError(f'Unknown speech encoder: {speech_encoder_type}')
|