egogpt/model/speech_encoder/builder.py

from .speech_encoder import WhisperWrappedEncoder


def build_speech_encoder(config):
    speech_encoder_type = getattr(config, "speech_encoder_type", None)
    if "whisper" in speech_encoder_type.lower():
        return WhisperWrappedEncoder(config)

    raise ValueError(f"Unknown speech encoder: {speech_encoder_type}")