torchaudio.models.wav2vec2.utils.import_huggingface_model

torchaudio.models.wav2vec2.utils.import_huggingface_model(original: Module) → Wav2Vec2Model

從 Transformers 中對應的模型物件構建 Wav2Vec2Model。

引數：

original (torch.nn.Module) – 來自 transformers 的 Wav2Vec2ForCTC 例項。

返回：

匯入的模型。

返回型別：

Wav2Vec2Model

示例

>>> from torchaudio.models.wav2vec2.utils import import_huggingface_model
>>>
>>> original = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
>>> model = import_huggingface_model(original)
>>>
>>> waveforms, _ = torchaudio.load("audio.wav")
>>> logits, _ = model(waveforms)