wav2vec2-ctc-ami / models /__init__.py
sanchit-gandhi's picture
Add scripts and weights
e984459
raw
history blame contribute delete
486 Bytes
from models.configuration_bart import BartConfig
from models.configuration_wav2vec2 import Wav2Vec2Config
from models.configuration_speech_encoder_decoder import SpeechEncoderDecoderConfig
from models.modeling_flax_wav2vec2 import FlaxWav2Vec2Model, FlaxWav2Vec2Module, FlaxWav2Vec2ForCTC, FlaxWav2Vec2ForCTCModule
from models.modeling_flax_bart import FlaxBartForCausalLM, FlaxBartForCausalLMModule
from models.modeling_flax_speech_encoder_decoder import FlaxSpeechEncoderDecoderModel