torchaudio/models/wav2vec2/__init__.py

from . import utils
from .model import (
    Wav2Vec2Model,
    HuBERTPretrainModel,
    wav2vec2_model,
    wav2vec2_base,
    wav2vec2_large,
    wav2vec2_large_lv60k,
    hubert_base,
    hubert_large,
    hubert_xlarge,
    hubert_pretrain_model,
    hubert_pretrain_base,
    hubert_pretrain_large,
    hubert_pretrain_xlarge,
)

__all__ = [
    "Wav2Vec2Model",
    "HuBERTPretrainModel",
    "wav2vec2_model",
    "wav2vec2_base",
    "wav2vec2_large",
    "wav2vec2_large_lv60k",
    "hubert_base",
    "hubert_large",
    "hubert_xlarge",
    "hubert_pretrain_model",
    "hubert_pretrain_base",
    "hubert_pretrain_large",
    "hubert_pretrain_xlarge",
    "utils",
]