• Zhaoheng Ni's avatar
    Add pre-trained pipelines for XLS-R models (#2978) · 9b7b64e4
    Zhaoheng Ni authored
    Summary:
    The PR adds three `Wav2Vec2Bundle ` pipeline objects for XLS-R models:
    - WAV2VEC2_XLSR_300M
    - WAV2VEC2_XLSR_1B
    - WAV2VEC2_XLSR_2B
    
    All three models use layer normalization in the feature extraction layers, hence `_normalize_waveform` is set to `True`.
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2978
    
    Reviewed By: hwangjeff
    
    Differential Revision: D42501491
    
    Pulled By: nateanl
    
    fbshipit-source-id: 2429ec880cc14798034843381e458e1b4664dac3
    9b7b64e4
refs.bib 22.9 KB