File size: 486 Bytes
af894aa |
1 2 3 4 5 6 7 |
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
|